Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ import time
|
|
5 |
import os
|
6 |
import random
|
7 |
|
8 |
-
os.environ["CUDA_VISIBLE_DEVICES"] = ""
|
9 |
|
10 |
# Create the gr.State component *outside* the gr.Blocks context
|
11 |
predictor_state = gr.State(None)
|
@@ -16,7 +16,7 @@ def get_transformer_model_id(task_type: str) -> str:
|
|
16 |
else:
|
17 |
return "Skywork/skyreels-v1-Hunyuan-t2v"
|
18 |
|
19 |
-
@spaces.GPU(duration=120)
|
20 |
def init_predictor(task_type: str):
|
21 |
# ALL IMPORTS NOW INSIDE THIS FUNCTION
|
22 |
import torch
|
|
|
5 |
import os
|
6 |
import random
|
7 |
|
8 |
+
# os.environ["CUDA_VISIBLE_DEVICES"] = ""
|
9 |
|
10 |
# Create the gr.State component *outside* the gr.Blocks context
|
11 |
predictor_state = gr.State(None)
|
|
|
16 |
else:
|
17 |
return "Skywork/skyreels-v1-Hunyuan-t2v"
|
18 |
|
19 |
+
# @spaces.GPU(duration=120)
|
20 |
def init_predictor(task_type: str):
|
21 |
# ALL IMPORTS NOW INSIDE THIS FUNCTION
|
22 |
import torch
|