Update app.py

app.py

@@ -25,8 +25,9 @@ import dlib
 num_bins = 256
 
 # Load the model
-model = load_model('best_model_HQ_v11.h5')
-model2 = load_model('best_model_HQ_v9.h5')
+# model = load_model('best_model_HQ_v11.h5')
+# model2 = load_model('best_model_HQ_v9.h5')
+model3 = load_model('best_model_HQ_enhanced_v12.h5')
 def u_sliding_factor(image_channel, P):
     result = np.zeros(image_channel.shape, np.float32)
 
@@ -233,16 +234,23 @@ if picture:
                         X_array = np.array(features_list)
                         # 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.85 and prediction2 >= 0.0000000000001:
-                            st.text("صورة حقيقية")
-                            # st.text(str(prediction[0]))
-                            # st.text(str(prediction2[0]))
+                        # prediction = model.predict(X_test_array_reshaped)
+                        # prediction2 = model2.predict(X_test_array_reshaped)
+                        prediction3 = model3.predict(X_test_array_reshaped)
+                        if prediction3 >=0.85:
+                            st.text("Real")
+                            st.text(str(prediction3[0]))
                         else:
-                            st.text("صورة مزيفة")
-                            st.text(str(prediction[0]))
-                            # st.text(str(prediction2[0]))
+                            st.text("Fack")
+                            st.text(str(prediction3[0]))
+                        # if prediction >= 0.85 and prediction2 >= 0.0000000000001:
+                        #     st.text("صورة حقيقية")
+                        #     # st.text(str(prediction[0]))
+                        #     # st.text(str(prediction2[0]))
+                        # else:
+                        #     st.text("صورة مزيفة")
+                        #     st.text(str(prediction[0]))
+                        #     # st.text(str(prediction2[0]))
             else:
                 st.text("لا يوجد وجه")
         else: