Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -109,7 +109,9 @@ class Pipeline:
|
|
| 109 |
text = ""
|
| 110 |
with torch.autocast(self.device):
|
| 111 |
for start, stop, frames in iter_clips(video_path):
|
| 112 |
-
|
|
|
|
|
|
|
| 113 |
frames = self.val_transform(frames).unsqueeze(0)
|
| 114 |
if self.device == 'cuda':
|
| 115 |
frames = frames.to(self.device).half()
|
|
@@ -129,7 +131,9 @@ class Pipeline:
|
|
| 129 |
)
|
| 130 |
for i in range(num_return_sequences):
|
| 131 |
generated_text_str = self.decode_one(generated_text_ids[i], self.tokenizer)
|
| 132 |
-
|
|
|
|
|
|
|
| 133 |
return text
|
| 134 |
|
| 135 |
interface = gr.Interface(
|
|
|
|
| 109 |
text = ""
|
| 110 |
with torch.autocast(self.device):
|
| 111 |
for start, stop, frames in iter_clips(video_path):
|
| 112 |
+
text_to_add = f"{'-'*30} Predictions From: {start:2.3f}-{stop:2.3f} seconds {'-'*30}\n"
|
| 113 |
+
print(text_to_add)
|
| 114 |
+
text += text_to_add
|
| 115 |
frames = self.val_transform(frames).unsqueeze(0)
|
| 116 |
if self.device == 'cuda':
|
| 117 |
frames = frames.to(self.device).half()
|
|
|
|
| 131 |
)
|
| 132 |
for i in range(num_return_sequences):
|
| 133 |
generated_text_str = self.decode_one(generated_text_ids[i], self.tokenizer)
|
| 134 |
+
text_to_add = '\t{}: {}\n'.format(i, generated_text_str)
|
| 135 |
+
print(text_to_add)
|
| 136 |
+
text += text_to_add
|
| 137 |
return text
|
| 138 |
|
| 139 |
interface = gr.Interface(
|