YiftachEde commited on
Commit
fd20b88
·
1 Parent(s): 12a6a7a

updated app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -248,7 +248,8 @@ class RefinerInterface:
248
  print("Initializing InstantMesh models...")
249
  self.pipeline, self.model, self.infer_config = load_models()
250
  print("InstantMesh models loaded!")
251
-
 
252
  def refine_model(self, input_image, prompt, steps=75, guidance_scale=7.5):
253
  """Main refinement function"""
254
  # Process image and get refined output
 
248
  print("Initializing InstantMesh models...")
249
  self.pipeline, self.model, self.infer_config = load_models()
250
  print("InstantMesh models loaded!")
251
+
252
+ @spaces.GPU(duration=65)
253
  def refine_model(self, input_image, prompt, steps=75, guidance_scale=7.5):
254
  """Main refinement function"""
255
  # Process image and get refined output