Spaces:
Running
on
Zero
Running
on
Zero
Update ui_manager.py
Browse files- ui_manager.py +9 -0
ui_manager.py
CHANGED
|
@@ -754,10 +754,19 @@ class UIManager:
|
|
| 754 |
self.inpainting_history.pop(0)
|
| 755 |
|
| 756 |
quality_score = result.get('quality_score', 0)
|
|
|
|
|
|
|
|
|
|
| 757 |
status = f"Inpainting complete in {elapsed:.1f}s"
|
| 758 |
if quality_score > 0:
|
| 759 |
status += f" | Quality: {quality_score:.0f}/100"
|
| 760 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 761 |
return (
|
| 762 |
result.get('combined_image'),
|
| 763 |
result.get('preview_image'),
|
|
|
|
| 754 |
self.inpainting_history.pop(0)
|
| 755 |
|
| 756 |
quality_score = result.get('quality_score', 0)
|
| 757 |
+
metadata = result.get('metadata', {})
|
| 758 |
+
actual_steps = metadata.get('num_inference_steps', num_steps)
|
| 759 |
+
|
| 760 |
status = f"Inpainting complete in {elapsed:.1f}s"
|
| 761 |
if quality_score > 0:
|
| 762 |
status += f" | Quality: {quality_score:.0f}/100"
|
| 763 |
|
| 764 |
+
# Notify user if steps were reduced (Spaces first-load scenario)
|
| 765 |
+
if actual_steps < num_steps:
|
| 766 |
+
status += f"\n⚠️ Steps reduced: {num_steps} → {actual_steps} (Spaces optimization)"
|
| 767 |
+
if actual_steps <= 7:
|
| 768 |
+
status += f"\n💡 First request - reduced quality. Click 'Generate' again for better results!"
|
| 769 |
+
|
| 770 |
return (
|
| 771 |
result.get('combined_image'),
|
| 772 |
result.get('preview_image'),
|