Spaces:
Running
on
Zero
Running
on
Zero
Update ui_manager.py
Browse files- 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=
|
| 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=
|
| 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],
|