Spaces:
Sleeping
Sleeping
Fix mask ratio
Browse files
app.py
CHANGED
|
@@ -131,7 +131,7 @@ def predict(mask_ratio:float, image:Image):
|
|
| 131 |
takes PIL image and return reconstructed image and predicted emotion label
|
| 132 |
"""
|
| 133 |
mask_ratio = float(mask_ratio)
|
| 134 |
-
masked_image, re_image = mae_reconstruct(image, figure_name='figure/demo_temp.png')
|
| 135 |
predicted_labels = classify(image)
|
| 136 |
|
| 137 |
return masked_image, re_image, predicted_labels
|
|
|
|
| 131 |
takes PIL image and return reconstructed image and predicted emotion label
|
| 132 |
"""
|
| 133 |
mask_ratio = float(mask_ratio)
|
| 134 |
+
masked_image, re_image = mae_reconstruct(image, figure_name='figure/demo_temp.png', mask_ratio=mask_ratio)
|
| 135 |
predicted_labels = classify(image)
|
| 136 |
|
| 137 |
return masked_image, re_image, predicted_labels
|