Update app.py

app.py

@@ -2,7 +2,6 @@
 import gradio as gr
 import numpy as np
 import tensorflow as tf
-import tensorflow_hub as hub
 import cv2
 
 # Load class labels from the text file
@@ -37,9 +36,9 @@ def predict_image(image):
     label = train_info[top_class]  # Get the corresponding label
     return label
 
-# Define the Gradio interface
-input_image = gr.inputs.Image(shape=(224, 224))
-output_label = gr.outputs.Label()
+# Updated Gradio interface
+input_image = gr.Image(shape=(224, 224))
+output_label = gr.Label()
 
 # Launch the Gradio interface
 gr.Interface(fn=predict_image, inputs=input_image, outputs=output_label, capture_session=True).launch()