Update app.py

app.py

@@ -93,8 +93,7 @@ with gr.Blocks() as demo:
             image_input = gr.Image(type="numpy", label="Upload Image")
             analyze_btn = gr.Button("Classify / Predict")
         
-        output_label = gr.Label(label="Prediction Scores")  # Output on the right
-
-    analyze_btn.click(fn=classify, inputs=[image_input], outputs=output_label)
+        output_label = gr.Label(label="Prediction Scores")
+        analyze_btn.click(fn=classify, inputs=[image_input, selected_model], outputs=output_label)
 
 demo.launch()