Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -105,7 +105,7 @@ class Pipeline:
|
|
105 |
generated_text_str = tokenizer.tokenizer.decode(generated_ids[1:eos_id].tolist())
|
106 |
return generated_text_str
|
107 |
|
108 |
-
def __call__(self, video_path, temperature=0.7, top_p=0.95, max_text_length=
|
109 |
text = ""
|
110 |
MAX_ITERATIONS = 5
|
111 |
with torch.autocast(self.device):
|
|
|
105 |
generated_text_str = tokenizer.tokenizer.decode(generated_ids[1:eos_id].tolist())
|
106 |
return generated_text_str
|
107 |
|
108 |
+
def __call__(self, video_path, temperature=0.7, top_p=0.95, max_text_length=120, num_return_sequences=5):
|
109 |
text = ""
|
110 |
MAX_ITERATIONS = 5
|
111 |
with torch.autocast(self.device):
|