David Fischinger
commited on
Commit
·
d7d7f68
1
Parent(s):
9a46ff1
bugfix: image conversion to PIL
Browse files
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/")
|
| 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 |
-
|
|
|
|
| 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 |
|