Update app.py

app.py

@@ -29,14 +29,14 @@ def predict(image):
     mobilenet_label = np.argmax(mobilenet_pred)
     custom_cnn_label = np.argmax(custom_cnn_pred)
     
-    return {
-        "VGG16 Prediction": vgg_label,
-        "MobileNet Prediction": mobilenet_label,
-        "Custom CNN Prediction": custom_cnn_label
-    }
+    return custom_cnn_label, vgg_label, mobilenet_label
 
 # Create the Gradio interface
 inputs = gr.components.Image()
-outputs = gr.components.Label(num_top_classes=3)
+outputs=[
+        gr.components.Textbox(label="Custom CNN Model Label"),
+        gr.components.Textbox(label="VGG16 Model Label"),
+        gr.components.Textbox(label="MobileNet Label")
+    ]
 
-gr.Interface(fn=predict, inputs=inputs, outputs=outputs).launch()
+gr.Interface(fn=predict, inputs=inputs, outputs=outputs, title="Fish Classification", theme="dark").launch()