Update app.py

app.py

@@ -1,7 +1,6 @@
 import tensorflow as tf
 import gradio as gr
 import numpy as np
-from PIL import Image
 
 # Load the model
 model = tf.saved_model.load('.')
@@ -18,8 +17,8 @@ def predict(image):
     return predictions.numpy().tolist()  # Adjust output processing as needed
 
 # Set up the Gradio interface
-image_input = gr.inputs.Image(type="pil")
-label_output = gr.outputs.Label(num_top_classes=3)
+image_input = gr.Image(type="pil")
+label_output = gr.Label(num_top_classes=3)
 
 interface = gr.Interface(fn=predict, inputs=image_input, outputs=label_output)
 interface.launch()