Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,6 +4,7 @@ import uuid
|
|
| 4 |
import tempfile
|
| 5 |
import subprocess
|
| 6 |
import torch
|
|
|
|
| 7 |
import torch.nn.functional as F
|
| 8 |
import torchaudio
|
| 9 |
import os
|
|
@@ -1452,13 +1453,13 @@ def get_duration(
|
|
| 1452 |
extra_time += 60
|
| 1453 |
|
| 1454 |
if duration <= 3:
|
| 1455 |
-
return
|
| 1456 |
elif duration <= 5:
|
| 1457 |
-
return
|
| 1458 |
elif duration <= 10:
|
| 1459 |
-
return
|
| 1460 |
else:
|
| 1461 |
-
return
|
| 1462 |
|
| 1463 |
@spaces.GPU(duration=get_duration)
|
| 1464 |
def generate_video(
|
|
|
|
| 4 |
import tempfile
|
| 5 |
import subprocess
|
| 6 |
import torch
|
| 7 |
+
print(f"Torch version: {torch.__version__}")
|
| 8 |
import torch.nn.functional as F
|
| 9 |
import torchaudio
|
| 10 |
import os
|
|
|
|
| 1453 |
extra_time += 60
|
| 1454 |
|
| 1455 |
if duration <= 3:
|
| 1456 |
+
return 60 + extra_time
|
| 1457 |
elif duration <= 5:
|
| 1458 |
+
return 80 + extra_time
|
| 1459 |
elif duration <= 10:
|
| 1460 |
+
return 120 + extra_time
|
| 1461 |
else:
|
| 1462 |
+
return 180 + extra_time
|
| 1463 |
|
| 1464 |
@spaces.GPU(duration=get_duration)
|
| 1465 |
def generate_video(
|