friesti1 commited on
Commit
d1e6dff
·
verified ·
1 Parent(s): 0c59e8e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ def predict_pokemon_type(uploaded_file):
10
  if uploaded_file is None:
11
  return "No file uploaded."
12
 
13
- model = tf.keras.models.load_model('Gengar-vs-Seel-vs-Zapdos-model_transferlearning.keras')
14
  # Load the image from the file path
15
  with Image.open(uploaded_file) as img:
16
  img = img.resize((150, 150)).convert('RGB') # Convert image to RGB
 
10
  if uploaded_file is None:
11
  return "No file uploaded."
12
 
13
+ model = tf.keras.models.load_model('pokemon-model_transferlearning.keras')
14
  # Load the image from the file path
15
  with Image.open(uploaded_file) as img:
16
  img = img.resize((150, 150)).convert('RGB') # Convert image to RGB