franckew commited on
Commit
2af03c7
·
verified ·
1 Parent(s): d39f85c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from tensorflow.keras.applications.resnet50 import preprocess_input
6
  from tensorflow.keras.models import load_model
7
 
8
  # Load your trained model
9
- model = load_model('/home/user/app/one_piece_character_classifier.h5') # Ensure this path is correct
10
 
11
  def predict_pokemon(img):
12
  img = Image.fromarray(img.astype('uint8'), 'RGB') # Ensure the image is in RGB
 
6
  from tensorflow.keras.models import load_model
7
 
8
  # Load your trained model
9
+ model = load_model('/home/user/app/franckew.h5') # Ensure this path is correct
10
 
11
  def predict_pokemon(img):
12
  img = Image.fromarray(img.astype('uint8'), 'RGB') # Ensure the image is in RGB