Update app.py

app.py

@@ -2,7 +2,6 @@ from PIL import Image
 import tensorflow as tf
 import numpy as np
 import gradio as gr
-from tensorflow.keras.models import load_model
 import io
 
 # Load the model
@@ -23,11 +22,11 @@ def preprocess_image(image):
 def predict_image(image):
     # Save the image to a file-like object
     image_bytes = io.BytesIO()
-    image.save(image_bytes, format="JPG")
+    image.save(image_bytes, format="JPEG")  # Change "JPG" to "JPEG"
     image_bytes.seek(0)  # Reset file pointer to start
 
     # Load the image from the file-like object
-    image = tf.keras.preprocessing.image.load_img(image_bytes, target_size=(256, 256))
+    image = tf.keras.preprocessing.image.load_img(image_bytes, target_size=(256, 256), color_mode="rgb")  # Specify color_mode="rgb"
 
     img_array = preprocess_image(image)
     outputs = model.predict(img_array)
@@ -42,10 +41,8 @@ def predict_image(image):
     
     return {"prediction": predicted_label}
 
-    #return json.dumps(prediction_dict, indent=1)
-
 # Create the interface
-input_interface = gr.Image(type = "pil")
+input_interface = gr.Image(type="pil")
 iface = gr.Interface(
     fn=predict_image,
     inputs=input_interface,