2025-07-31 20:34 🐛

Fixed a bug in app.py

app.py

@@ -242,7 +242,7 @@ def predict(image: Image.Image, variant_dataset: str, metric: str):
     compute the total crowd count, and prepare the density map for display.
     """
     global loaded_model
-    variant, dataset = variant_dataset.value.split(" @ ")
+    variant, dataset = variant_dataset.split(" @ ")
 
     if loaded_model is None:
 
@@ -374,7 +374,7 @@ def predict(image: Image.Image, variant_dataset: str, metric: str):
     lambda_map = Image.blend(image_rgba, lambda_map, alpha=alpha)
     complete_zero_map = Image.blend(image_rgba, complete_zero_map, alpha=alpha)
     
-    return image, strucrual_zero_map, sampling_zero_map, complete_zero_map, lambda_map, den_map, f"Predicted Count: {count:.2f}"
+    return image, den_map, lambda_map, round(count, 2), strucrual_zero_map, sampling_zero_map, complete_zero_map
 
 
 # -----------------------------
@@ -405,17 +405,18 @@ with gr.Blocks() as demo:
     
         with gr.Column():
             output_den_map = gr.Image(label="Predicted Density Map", type="pil")
+            output_lambda_map = gr.Image(label="Lambda Map", type="pil")
+            output_text = gr.Textbox(label="Predicted Count")
+        
+        with gr.Column():
             output_structural_zero_map = gr.Image(label="Structural Zero Map", type="pil")
             output_sampling_zero_map = gr.Image(label="Sampling Zero Map", type="pil")
-            output_lambda_map = gr.Image(label="Lambda Map", type="pil")
             output_complete_zero_map = gr.Image(label="Complete Zero Map", type="pil")
-            
-            output_text = gr.Textbox(label="Total Count")
 
     submit_btn.click(
         fn=predict,
         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]
+        outputs=[input_img, output_den_map, output_lambda_map, output_text, output_structural_zero_map, output_sampling_zero_map, output_complete_zero_map]
     )
 
     gr.Examples(