Hugging Face's logo

Spaces:
AliSaria
/
MilitarEye
Sleeping

App Files Community
AliSaria commited on
Commit
6774ffb
·
verified ·
1 Parent(s): d66ae1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import matplotlib.pyplot as plt
6
  from io import BytesIO
7
 
8
  # Load the trained model
9
- model = load_model('model1.h5') # Make sure 'model1.h5' is the correct path to your model
10
 
11
  # Prediction function for the Gradio app
12
  def predict_and_visualize(img):
 
6
  from io import BytesIO
7
 
8
  # Load the trained model
9
+ model = load_model('model2.h5') # Make sure 'model1.h5' is the correct path to your model
10
 
11
  # Prediction function for the Gradio app
12
  def predict_and_visualize(img):