Spaces:
Sleeping
Sleeping
Fix image output order to match UI layout
Browse files- Swap loss_plot and super_resolved in return statement
- Now correctly shows: downsampled+loss (row 1), prediction+GT (row 2)
app.py
CHANGED
|
@@ -151,7 +151,8 @@ def super_resolve_image(input_image, scale_factor, training_steps, hidden_featur
|
|
| 151 |
plt.close()
|
| 152 |
|
| 153 |
# Return individual images and metrics
|
| 154 |
-
|
|
|
|
| 155 |
|
| 156 |
|
| 157 |
# Create Gradio interface
|
|
|
|
| 151 |
plt.close()
|
| 152 |
|
| 153 |
# Return individual images and metrics
|
| 154 |
+
# Order: downsampled, loss_plot, super_resolved, gt, metrics (matches UI layout)
|
| 155 |
+
return downsampled_image, loss_plot, super_resolved_image, gt_image, metrics_text
|
| 156 |
|
| 157 |
|
| 158 |
# Create Gradio interface
|