royeis commited on
Commit
c47cc24
·
1 Parent(s): f399951

increase loading spaces gpu duration

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -90,7 +90,7 @@ def reset_progress_tracking():
90
  baseline_max_progress = 0.0
91
  logging.info("Progress tracking variables reset for new generation")
92
 
93
- @spaces.GPU()
94
  def load_model_and_rnn():
95
  """Load both model and RNN model with hardcoded defaults"""
96
  model_name = "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B"
 
90
  baseline_max_progress = 0.0
91
  logging.info("Progress tracking variables reset for new generation")
92
 
93
+ @spaces.GPU(duration=120)
94
  def load_model_and_rnn():
95
  """Load both model and RNN model with hardcoded defaults"""
96
  model_name = "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B"