ruslanmv commited on
Commit
bcd53e1
·
verified ·
1 Parent(s): e74409f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -84,7 +84,7 @@ check_gpu_availability()
84
  # GPU-Safe MinDalle Model Loading
85
  def initialize_min_dalle_with_gpu():
86
  """Load the MinDalle model with GPU support."""
87
- @spaces.GPU(duration=60 * 3)
88
  def load_model():
89
  print("Loading MinDalle model...")
90
  return MinDalle(
 
84
  # GPU-Safe MinDalle Model Loading
85
  def initialize_min_dalle_with_gpu():
86
  """Load the MinDalle model with GPU support."""
87
+ @spaces.GPU()
88
  def load_model():
89
  print("Loading MinDalle model...")
90
  return MinDalle(