Update app.py

app.py

@@ -35,7 +35,7 @@ drawing_spec = mp_drawing.DrawingSpec(thickness=1, circle_radius=1)
 
 # Load the model
 model = load_model('best_model_HQ_v11.h5')
-# model2 = load_model('best_model_HQ_v9.h5')
+model2 = load_model('best_model_HQ_v9.h5')
 def u_sliding_factor(image_channel, P):
     result = np.zeros(image_channel.shape, np.float32)
 
@@ -219,8 +219,8 @@ if picture:
                 # print(X_array.shape)
                 X_test_array_reshaped = np.expand_dims(X_array, axis=-1)
                 prediction = model.predict(X_test_array_reshaped)
-                # prediction2 = model2.predict(X_test_array_reshaped)
-                if prediction >= 0.90:
+                prediction2 = model2.predict(X_test_array_reshaped)
+                if prediction >= 0.90 and prediction2 >= 0.0000000000001:
                     st.text("صورة حقيقية")
                     st.text(str(prediction[0]))
                     # st.text(str(prediction2[0]))