EngAbod commited on
Commit
a2aed3f
·
1 Parent(s): bec9e40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -34,8 +34,8 @@ mp_drawing = mp.solutions.drawing_utils
34
  drawing_spec = mp_drawing.DrawingSpec(thickness=1, circle_radius=1)
35
 
36
  # Load the model
37
- model = load_model('best_model_HQ_v8.h5')
38
- model2 = load_model('best_model_HQ_v9.h5')
39
  def u_sliding_factor(image_channel, P):
40
  result = np.zeros(image_channel.shape, np.float32)
41
 
@@ -219,8 +219,8 @@ if picture:
219
  # print(X_array.shape)
220
  X_test_array_reshaped = np.expand_dims(X_array, axis=-1)
221
  prediction = model.predict(X_test_array_reshaped)
222
- prediction2 = model2.predict(X_test_array_reshaped)
223
- if prediction >= 0.00001 and prediction2 >= 0.000001:
224
  st.text("صورة حقيقية")
225
  # st.text(str(prediction[0]))
226
  # st.text(str(prediction2[0]))
 
34
  drawing_spec = mp_drawing.DrawingSpec(thickness=1, circle_radius=1)
35
 
36
  # Load the model
37
+ model = load_model('best_model_HQ_v10.h5')
38
+ # model2 = load_model('best_model_HQ_v9.h5')
39
  def u_sliding_factor(image_channel, P):
40
  result = np.zeros(image_channel.shape, np.float32)
41
 
 
219
  # print(X_array.shape)
220
  X_test_array_reshaped = np.expand_dims(X_array, axis=-1)
221
  prediction = model.predict(X_test_array_reshaped)
222
+ # prediction2 = model2.predict(X_test_array_reshaped)
223
+ if prediction >= 0.1:
224
  st.text("صورة حقيقية")
225
  # st.text(str(prediction[0]))
226
  # st.text(str(prediction2[0]))