1inkusFace commited on
Commit
bb1eaa6
·
verified ·
1 Parent(s): f894bb5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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