David Fischinger
commited on
Commit
·
6c58fe6
1
Parent(s):
4d61a69
fix versions in requirements.txt
Browse files- app.py +2 -2
- requirements.txt +6 -5
app.py
CHANGED
|
@@ -27,8 +27,8 @@ 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 |
|
|
|
|
| 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/model1.h5")
|
| 31 |
+
model_M2 = tf.keras.models.load_model("IMVIP_Supplementary_Material/models/model2/model2.h5")
|
| 32 |
|
| 33 |
return model_M1, model_M2
|
| 34 |
|
requirements.txt
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
-
opencv-python
|
| 2 |
-
tensorflow
|
| 3 |
-
pillow
|
| 4 |
-
scikit-learn
|
| 5 |
-
|
|
|
|
|
|
| 1 |
+
opencv-python==4.10.0.84
|
| 2 |
+
tensorflow==2.12.0
|
| 3 |
+
pillow==9.5.0
|
| 4 |
+
scikit-learn==1.5.2
|
| 5 |
+
matplotlib==3.9.2
|
| 6 |
+
protobuf==3.20.3
|