Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
| 29 |
predictor = None # Global predictor, BUT loaded inside a function
|
| 30 |
|
| 31 |
def get_transformer_model_id(task_type: str) -> str:
|
| 32 |
-
return "Skywork/
|
| 33 |
|
| 34 |
@spaces.GPU(duration=90)
|
| 35 |
def init_predictor(task_type: str):
|
|
|
|
| 29 |
predictor = None # Global predictor, BUT loaded inside a function
|
| 30 |
|
| 31 |
def get_transformer_model_id(task_type: str) -> str:
|
| 32 |
+
return "Skywork/skyreels-v1-Hunyuan-i2v" if task_type == "i2v" else "Skywork/skyreels-v1-Hunyuan-t2v"
|
| 33 |
|
| 34 |
@spaces.GPU(duration=90)
|
| 35 |
def init_predictor(task_type: str):
|