Update app.py

app.py

@@ -2,6 +2,8 @@ import torch
 from torchvision import transforms
 from PIL import Image
 from torchvision import models
+import gradio.inputs as gi
+import gradio.outputs as go
 import gradio as gr
 
 # Define the ResNet50 model
@@ -45,7 +47,7 @@ def predict_image_label(image):
     return label
 
 # Create Gradio interface
-image = gr.inputs.Image(shape=(224, 224))
-label = gr.outputs.Label(num_top_classes=2)
+image = gi.Image(shape=(224, 224))
+label = go.Label(num_top_classes=2)
 
 gr.Interface(fn=predict_image_label, inputs=image, outputs=label, title="Leaf or Plant Classifier").launch()