Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -314,6 +314,7 @@ def extract_audio_from_video(video_path, audio_path=None):
|
|
| 314 |
|
| 315 |
import time
|
| 316 |
|
|
|
|
| 317 |
def generate_text(video_path, audio_track, prompt):
|
| 318 |
streamer = TextIteratorStreamer(tokenizer, skip_special_tokens=True)
|
| 319 |
|
|
|
|
| 314 |
|
| 315 |
import time
|
| 316 |
|
| 317 |
+
@spaces.GPU
|
| 318 |
def generate_text(video_path, audio_track, prompt):
|
| 319 |
streamer = TextIteratorStreamer(tokenizer, skip_special_tokens=True)
|
| 320 |
|