Spaces:
Running
Running
Increase gpu allocation
Browse files
app.py
CHANGED
|
@@ -398,7 +398,7 @@ def extract_video_landmarks(video_frames, landmarks_extractor):
|
|
| 398 |
return np.array(processed_landmarks)
|
| 399 |
|
| 400 |
|
| 401 |
-
@spaces.GPU(duration=
|
| 402 |
@torch.no_grad()
|
| 403 |
def sample(
|
| 404 |
audio_list,
|
|
@@ -617,7 +617,7 @@ def sample(
|
|
| 617 |
return complete_video
|
| 618 |
|
| 619 |
|
| 620 |
-
@spaces.GPU(duration=
|
| 621 |
@torch.no_grad()
|
| 622 |
def process_video(video_input, audio_input):
|
| 623 |
"""Main processing function to generate synchronized video"""
|
|
|
|
| 398 |
return np.array(processed_landmarks)
|
| 399 |
|
| 400 |
|
| 401 |
+
@spaces.GPU(duration=200)
|
| 402 |
@torch.no_grad()
|
| 403 |
def sample(
|
| 404 |
audio_list,
|
|
|
|
| 617 |
return complete_video
|
| 618 |
|
| 619 |
|
| 620 |
+
@spaces.GPU(duration=200)
|
| 621 |
@torch.no_grad()
|
| 622 |
def process_video(video_input, audio_input):
|
| 623 |
"""Main processing function to generate synchronized video"""
|