Update app.py

Fix the summary logic

app.py

@@ -42,12 +42,7 @@ def detect_objects(image_url, uploaded_image, text_prompt):
             return None, "Please provide an image URL or upload an image."
 
         # Default prompt fallback
-        if not text_prompt or not text_prompt.strip():
-            text_prompt = "capsule"
-        
-        text_prompt = normalize_prompt(text_prompt)
-
-        display_prompt = text_prompt.strip() if text_prompt else "capsule"
+        display_prompt = text_prompt.strip() if text_prompt and text_prompt.strip() else "capsule"
         model_prompt = normalize_prompt(display_prompt)
 
         # Inference
@@ -59,7 +54,7 @@ def detect_objects(image_url, uploaded_image, text_prompt):
         results = processor.post_process_grounded_object_detection(
             outputs,
             inputs.input_ids,
-            threshold=0.35,
+            threshold=0.34,
             target_sizes=[image.size[::-1]]
         )
 
@@ -101,17 +96,28 @@ def detect_objects(image_url, uploaded_image, text_prompt):
                 """, image
 
         counts = Counter(detected_labels)
-        total = sum(counts.values())
 
         summary_rows = []
         
         for label, count in counts.items():
-            object_name = label if count == 1 else label + "s"
             summary_rows.append(
-                f"<tr><td style='padding:4px 12px'>{object_name}</td><td style='padding:4px 12px'><b>{count}</b></td></tr>"
+                f"<tr><td style='padding:4px 12px'>{label}</td>"
+                f"<td style='padding:4px 12px'><b>{count}</b></td></tr>"
             )
         
-        summary = f"<h3>Detected {total} object(s) for: {display_prompt}</h3>"
+        total_types = len(counts)
+        
+        summary = f"""
+            <h3>Detected {total_types} object type(s) for: {display_prompt}</h3>
+            
+            <table style='border-collapse: collapse; width: 100%;'>
+            <tr>
+            <th style='text-align:left; padding:4px 12px;'>Object</th>
+            <th style='text-align:left; padding:4px 12px;'>Count</th>
+            </tr>
+            {''.join(summary_rows)}
+            </table>
+            """
 
         return summary, image