bennyguo commited on
Commit
854df12
·
1 Parent(s): 1acc996

remove the duration for zerogpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -153,7 +153,7 @@ def get_random_seed():
153
  return random.randint(0, MAX_SEED)
154
 
155
  # Apply decorator conditionally
156
- @spaces.GPU(duration=120) if ENABLE_ZEROGPU else lambda func: func
157
  def generate_3d(scribble_image_dict, prompt, scribble_confidence, seed): # Added seed parameter back
158
  print("Generating 3D model...")
159
  # Extract the composite image from the ImageEditor dictionary
 
153
  return random.randint(0, MAX_SEED)
154
 
155
  # Apply decorator conditionally
156
+ @spaces.GPU() if ENABLE_ZEROGPU else lambda func: func
157
  def generate_3d(scribble_image_dict, prompt, scribble_confidence, seed): # Added seed parameter back
158
  print("Generating 3D model...")
159
  # Extract the composite image from the ImageEditor dictionary