Upload 5 files

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+Negative.jpg filter=lfs diff=lfs merge=lfs -text
+Positive.jpg filter=lfs diff=lfs merge=lfs -text

ART_Positive_Negative_Detection_v1.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:42af7ad577a211f82b0c2c6d8afef8b2654c3291d81ddca93d13857a782ffab6
+size 52034945

app.py

@@ -0,0 +1,76 @@
+import gradio as gr
+from PIL import Image
+from ultralytics import YOLO
+import requests
+import json
+import logging
+
+logging.basicConfig(level=logging.INFO)
+
+model = YOLO("ART_Positive_Negative_Detection_v1.pt")
+
+def detect_objects(images):
+    results = model(images)
+    classes={ 0: "Positive", 1: "Negative"}
+    names = []
+    results = model(images, conf=0.3)  
+    for result in results:
+        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 in zip(image_urls, names):      
+        obj = {"url": image_url, "answer": [class_name] }   
+        solutions.append(obj)
+    return solutions
+
+# def send_results_to_api(data, result_url):
+#     # Example function to send results to an API
+#     headers = {"Content-Type": "application/json"}
+#     response = requests.post(result_url, json=data, headers=headers)
+#     if response.status_code == 200:
+#         return response.json()  # Return any response from the API if needed
+#     else:
+#         return {"error": f"Failed to send results to API: {response.status_code}"}
+
+def process_images(params):
+    try:
+        params = json.loads(params)
+    except json.JSONDecodeError as e:
+        logging.error(f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}")
+        return {"error": f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}"}
+    
+    image_urls = params.get("urls", [])
+    # api = params.get("api", "")
+    # job_id = params.get("job_id", "")
+    
+    if not image_urls:
+        logging.error("Missing required parameters: 'urls'")
+        return {"error": "Missing required parameters: 'urls'"}
+
+    try:
+        images = [Image.open(requests.get(url, stream=True).raw) for url in image_urls]  # images from URLs
+    except Exception as e:
+        logging.error(f"Error loading images: {e}")
+        return {"error": f"Error loading images: {str(e)}"}
+        
+    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)
+
+    return json.dumps({"solutions": solutions})
+
+inputt = gr.Textbox(label="Parameters (JSON format) Eg. img_url:['','']")
+outputs = gr.JSON()
+
+application = gr.Interface(fn=process_images, inputs=inputt, outputs=outputs, title="ART +ve -ve Detection with API Integration")
+application.launch()

requirements.txt

@@ -0,0 +1 @@
+ultralytics