Update app.py

app.py

@@ -12,9 +12,10 @@ def predict_pokemon_type(uploaded_file):
     model = tf.keras.models.load_model('biomes-xception-model.keras')
 
     # Load the image from the file path
-    with Image.open(uploaded_file) as img:
+    with Image.open(uploaded_file).convert('RGB') as img:       
         img = img.resize((150, 150))
         img_array = np.array(img)
+        img_array = (img_array / 127.5) - 1.0  # Rescale the image as the model expects
 
         prediction = model.predict(np.expand_dims(img_array, axis=0))