Update app.py

app.py

@@ -8,19 +8,19 @@ model = YOLO("ART_Positive_Negative_Detection_v1.pt")
 
 def detect_objects(images):
     results = model(images)
-    classes={0:"positive", 1:"negative"}
-    names=[]
-    probss=[]
+    classes={ 0: "Positive", 1: "Negative"}
+    names = []
+    results = model(images, conf=0.3)  
     for result in results:
-        probs = result.probs.top1  
-        probss.append(probs)
-        
-        arr=[]
-        arr.append(classes[probs])
-        names.append(arr)
-    return names, probss
-
-def create_solutions(image_urls, names, probss):
+        if len(result.boxes) > 0:
+            class_index = result.boxes.cls[0].item()
+            class_name = classes[class_index]
+            names.append([class_name])
+        else:
+            names.append(['None'])
+    return names
+
+def create_solutions(image_urls, names):
     solutions = []   #list to store all the objects
 
     for image_url, class_name, prob in zip(image_urls, names, probss):      
@@ -47,8 +47,8 @@ def process_images(params):
 
     images = [Image.open(requests.get(url, stream=True).raw) for url in image_urls]  # images from URLs
 
-    names, probss = detect_objects(images)  # Perform object detection
-    solutions = create_solutions(image_urls, names, probss)  # Create solutions with image URLs and bounding boxes
+    names = detect_objects(images)  # Perform object detection
+    solutions = create_solutions(image_urls, names)  # Create solutions with image URLs and bounding boxes
 
     result_url = f"{api}/{job_id}"
     # send_results_to_api(solutions, result_url)