Update app.py

app.py

@@ -44,13 +44,12 @@ def predict_class(image):
 # UI Design
 def classify_image(image):
     predicted_class = predict_class(image)
-    output = f"<h2>Predicted Class:</h2><p>{predicted_class}</p>"
-    return output
+    return image, predicted_class
 
 inputs = gr.inputs.Image(label="Upload an image")
-outputs = gr.outputs.HTML()
+outputs = gr.outputs.Image(label="Uploaded Image"), gr.outputs.Textbox(label="Predicted Class", live=True)
 
-title = "<h1 style='text-align: center;'>Image Classifier</h1>"
+title = "Image Classifier"
 description = "Upload an image and get the predicted class."
 
-gr.Interface(fn=classify_image, inputs=inputs, outputs=outputs, title=title, description=description).launch()
+gr.Interface(fn=classify_image, inputs=inputs, outputs=outputs, title=title, description=description).launch(inline=True)