Nipun commited on
Commit
c58ff0d
·
1 Parent(s): 691ba3c

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)

Files changed (1) hide show
  1. app.py +2 -1
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
- return downsampled_image, super_resolved_image, gt_image, loss_plot, metrics_text
 
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