Update app.py

app.py

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