2025-07-31 19:48 🐛

Fixed a bug in app.py

app.py

@@ -236,12 +236,13 @@ def _sliding_window_predict(
 # Inference function
 # -----------------------------
 @spaces.GPU(duration=120)
-def predict(image: Image.Image, variant: str, dataset: str, metric: str):
+def predict(image: Image.Image, variant_dataset: str, metric: str):
     """
     Given an input image, preprocess it, run the model to obtain a density map,
     compute the total crowd count, and prepare the density map for display.
     """
     global loaded_model
+    variant, dataset = variant_dataset.value.split(" @ ")
 
     if loaded_model is None:
 
@@ -411,10 +412,9 @@ with gr.Blocks() as demo:
             
             output_text = gr.Textbox(label="Total Count")
 
-    variant, dataset = model_dropdown.value.split(" @ ")
     submit_btn.click(
         fn=predict,
-        inputs=[input_img, variant, dataset, metric_dropdown],
+        inputs=[input_img, model_dropdown, metric_dropdown],
         outputs=[input_img, output_structural_zero_map, output_sampling_zero_map, output_complete_zero_map, output_lambda_map, output_den_map, output_text]
     )