David Fischinger commited on
Commit
d7d7f68
·
1 Parent(s): 9a46ff1

bugfix: image conversion to PIL

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -27,9 +27,9 @@ def load_models():
27
  model_path1 = "IMVIP_Supplementary_Material/models/model1/"
28
  model_path2 = "IMVIP_Supplementary_Material/models/model2/"
29
 
30
- model_M1 = tf.keras.models.load_model("IMVIP_Supplementary_Material/models/model1/") #tf.keras.models.load_model("IMVIP_Supplementary_Material/models/model1/")
31
  model_M2 = tf.keras.models.load_model("IMVIP_Supplementary_Material/models/model2/")
32
-
33
  return model_M1, model_M2
34
 
35
 
@@ -110,5 +110,6 @@ def reset_image_select():
110
  uploaded_file= None
111
  uploaded_file = st.file_uploader("Please upload an image", type=["jpeg", "jpg", "png"], on_change=reset_image_select)
112
  if (uploaded_file is not None) and (img is None):
113
- start_evaluation(uploaded_file)
 
114
 
 
27
  model_path1 = "IMVIP_Supplementary_Material/models/model1/"
28
  model_path2 = "IMVIP_Supplementary_Material/models/model2/"
29
 
30
+ model_M1 = tf.keras.models.load_model("IMVIP_Supplementary_Material/models/model1/")
31
  model_M2 = tf.keras.models.load_model("IMVIP_Supplementary_Material/models/model2/")
32
+
33
  return model_M1, model_M2
34
 
35
 
 
110
  uploaded_file= None
111
  uploaded_file = st.file_uploader("Please upload an image", type=["jpeg", "jpg", "png"], on_change=reset_image_select)
112
  if (uploaded_file is not None) and (img is None):
113
+ pil_img = Image.open(uploaded_file)
114
+ start_evaluation(pil_img)
115