nateraw commited on
Commit
ecd6abf
β€’
1 Parent(s): 76f35cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=77, num_return_sequences=10):
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):