Spaces:
Running
on
Zero
Running
on
Zero
BoboiAzumi
commited on
Commit
·
af76564
1
Parent(s):
11ad142
fix
Browse files- process.py +1 -1
process.py
CHANGED
@@ -104,7 +104,7 @@ def predict(prompt, video_data, temperature):
|
|
104 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
105 |
return response
|
106 |
|
107 |
-
def
|
108 |
temperature = 0.1
|
109 |
video_data = open(video, 'rb').read()
|
110 |
response = predict(prompt, video_data, temperature)
|
|
|
104 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
105 |
return response
|
106 |
|
107 |
+
def inference(video, prompt):
|
108 |
temperature = 0.1
|
109 |
video_data = open(video, 'rb').read()
|
110 |
response = predict(prompt, video_data, temperature)
|