Update app.py

app.py

@@ -135,28 +135,30 @@ def make_prediction(image, choose_resolution, taxonomic_level):
     else:
         return make_prediction_auto(image)
 
-# Define the Gradio interface
-interface = gr.Interface(
-    fn=make_prediction,            
-    inputs=[
-        gr.Image(type="pil"),  
-        gr.Radio(choices=["Yes, I want to specify the taxonomic level", "No, I will let the model decide"], 
-                 label="Do you want to choose the taxonomic resolution for predictions?", value="No, I will let the model decide"),
-        gr.Dropdown(choices=taxonomic_levels, label="Taxonomic level", value="species", interactive=True)],  
-    outputs="html",                
-    title="Amazon arboreal species classification",
-    description="Upload an image and select the taxonomic level (optional) to classify the species."
-)
-
-# Add custom logic to disable the "Taxonomic level" dropdown when "No, I will let the model decide" is selected
+# Function to dynamically disable/enable the taxonomic level dropdown
 def update_taxonomic_level_interface(choose_resolution):
     if choose_resolution == "No, I will let the model decide":
         return gr.Dropdown.update(interactive=False)
     else:
         return gr.Dropdown.update(interactive=True)
 
-# Set up dynamic behavior for the interface
-interface.update(inputs=["Do you want to choose the taxonomic resolution for predictions?"], fn=update_taxonomic_level_interface)
+# Define the Gradio interface
+with gr.Blocks() as interface:
+    with gr.Row():
+        image_input = gr.Image(type="pil")
+        choose_resolution = gr.Radio(
+            choices=["Yes, I want to specify the taxonomic level", "No, I will let the model decide"], 
+            label="Do you want to choose the taxonomic resolution for predictions?", value="No, I will let the model decide"
+        )
+        taxonomic_level = gr.Dropdown(
+            choices=taxonomic_levels, label="Taxonomic level", value="species", interactive=True
+        )
+    
+    result_output = gr.HTML()
+
+    # Bind the prediction function to the inputs
+    choose_resolution.change(fn=update_taxonomic_level_interface, inputs=choose_resolution, outputs=taxonomic_level)
+    image_input.submit(fn=make_prediction, inputs=[image_input, choose_resolution, taxonomic_level], outputs=result_output)
 
 # Launch the Gradio interface
 interface.launch()