Update app.py

app.py

@@ -13,10 +13,14 @@ class_names = ['CNV','DME','DRUSEN','NORMAL']
 
 
 # 2. Model annd transforms prepration
+# model = tf.keras.models.load_model(
+#     'oct_classification_final_model_lg.keras', custom_objects=None, compile=True, safe_mode=True
+# )
 model = tf.keras.models.load_model(
-    'oct_classification_final_model_lg.keras', custom_objects=None, compile=True, safe_mode=True
+    'oct_classification_final_model_lg.keras', custom_objects=None, compile=True, safe_mode=False
 )
 
+
 # Load save weights
 
 # 3.prediction function (predict())