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):
|