Spaces:
Running
on
Zero
Running
on
Zero
Update hf_gradio_app.py
Browse files- hf_gradio_app.py +1 -1
hf_gradio_app.py
CHANGED
|
@@ -100,7 +100,7 @@ def process_audio(file_path, temp_dir):
|
|
| 100 |
return output_path
|
| 101 |
|
| 102 |
#@torch.inference_mode()
|
| 103 |
-
@spaces.GPU()
|
| 104 |
def generate(input_video, input_audio, seed, progress=gr.Progress(track_tqdm=True)):
|
| 105 |
pipeline.to(device=device, dtype=weight_dtype)
|
| 106 |
|
|
|
|
| 100 |
return output_path
|
| 101 |
|
| 102 |
#@torch.inference_mode()
|
| 103 |
+
@spaces.GPU(duration=240)
|
| 104 |
def generate(input_video, input_audio, seed, progress=gr.Progress(track_tqdm=True)):
|
| 105 |
pipeline.to(device=device, dtype=weight_dtype)
|
| 106 |
|