Spaces:
Running
Running
fix to return correct img
Browse files
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,
|
| 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)
|