ryu03153 commited on
Commit
87a0a2f
·
verified ·
1 Parent(s): ab2b6d5

fix to return correct img

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -100,7 +100,7 @@ def mae_reconstruct(image:Image, figure_name='figure/demo_temp.png', mask_ratio=
100
  recon_np = to_img(img_rec)
101
  hybrid_np = to_img(img_hybrid)
102
 
103
- return masked_np, recon_np
104
 
105
  def classify(image:Image):
106
  img = transform(image).unsqueeze(0)
 
100
  recon_np = to_img(img_rec)
101
  hybrid_np = to_img(img_hybrid)
102
 
103
+ return masked_np, hybrid_np
104
 
105
  def classify(image:Image):
106
  img = transform(image).unsqueeze(0)