Update app.py

app.py

@@ -94,7 +94,7 @@ def fit_model(zip_file_path):
 
 def predict(image):
     l = m.predict(image)
-    return f'Predicted Class: {l}'
+    return f'Predicted Class: {l[0]}'
 
 
 def export(password):
@@ -112,18 +112,18 @@ with gr.Blocks() as demo:
 
     with gr.Row() as g1:
         inp = gr.File(label="zip file")
-        out = gr.Textbox()
+        out = gr.Textbox(label='Message')
     btn = gr.Button("Submit Zip File")
     btn.click(fn=fit_model, inputs=inp, outputs=out)
 
     with gr.Row() as g2:
         inp2 = gr.Image(label="Input Image", type='pil')
-        out2 = gr.Textbox()
+        out2 = gr.Textbox(label='Prediction')
     btn2 = gr.Button("Predict/Test on an Image")
     btn2.click(fn=predict, inputs=inp2, outputs=out2)
 
     with gr.Row() as g3:
-        inp3 = gr.Textbox()
+        inp3 = gr.Textbox(label='Password to Download')
         out3 = gr.File(label='Download link', visible=True, height=30, interactive=False)
         out4 = gr.File(label='Download link', visible=True, height=30, interactive=False)
         out5 = gr.File(label='Download link', visible=True, height=30, interactive=False)