DawnC commited on
Commit
94367e1
·
verified ·
1 Parent(s): aa7bf19

Update ui_manager.py

Browse files
Files changed (1) hide show
  1. ui_manager.py +2 -2
ui_manager.py CHANGED
@@ -173,7 +173,7 @@ class UIManager:
173
  if len(self.generation_history) > max_history:
174
  self.generation_history = self.generation_history[-max_history:]
175
 
176
- @spaces.GPU(duration=120)
177
  def generate_handler(
178
  self,
179
  uploaded_image: Optional[Image.Image],
@@ -652,7 +652,7 @@ class UIManager:
652
  logger.error(f"Failed to extract mask from editor: {e}")
653
  return None
654
 
655
- @spaces.GPU(duration=120)
656
  def inpainting_handler(
657
  self,
658
  image: Optional[Image.Image],
 
173
  if len(self.generation_history) > max_history:
174
  self.generation_history = self.generation_history[-max_history:]
175
 
176
+ @spaces.GPU(duration=240)
177
  def generate_handler(
178
  self,
179
  uploaded_image: Optional[Image.Image],
 
652
  logger.error(f"Failed to extract mask from editor: {e}")
653
  return None
654
 
655
+ @spaces.GPU(duration=300)
656
  def inpainting_handler(
657
  self,
658
  image: Optional[Image.Image],