prekshyam commited on
Commit
122057c
·
verified ·
1 Parent(s): 7e3569f

Fix mask ratio

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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