Update app.py

app.py

@@ -81,8 +81,8 @@ def predict(image):
 # Set up Gradio interface
 interface = gr.Interface(
     fn=predict,
-    inputs=gr.inputs.Image(type="pil"),
-    outputs="text",
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Text(label="Prediction"),
     title="Bacterial Morphology Classification",
     description="Upload an image of bacteria to classify it as cocci, bacilli, or spirilla.",
 )