D3V1L1810 commited on
Commit
9c8d143
·
verified ·
1 Parent(s): b286021

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -23,10 +23,10 @@ def detect_objects(images):
23
  names.append(arr)
24
  return names, probss
25
 
26
- def create_solutions(image_urls, names, probss,normalFileID):
27
  solutions = [] #list to store all the objects
28
 
29
- for image_url, class_name, prob, fileID in zip(image_urls, names, probss,normalFileID):
30
  obj = {"image": image_url, "answer": [class_name], "qcUser" : None}
31
  solutions.append(obj)
32
  print("image done")
@@ -49,7 +49,7 @@ def process_images(params):
49
  return {"error": f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}"}
50
 
51
  image_urls = params.get("urls", [])
52
- normalFileId = 0
53
  # normalFileId = params.get("normalfileID",[])
54
  # api = params.get("api", "")
55
  # job_id = params.get("job_id", "")
@@ -65,7 +65,7 @@ def process_images(params):
65
  return {"error": f"Error loading images: {str(e)}"}
66
 
67
  names, probss = detect_objects(images) # Perform object detection
68
- solutions = create_solutions(image_urls, names, probss,normalFileId) # Create solutions with image URLs and bounding boxes
69
 
70
  # result_url = f"{api}/{job_id}"
71
  # send_results_to_api(solutions, result_url)
 
23
  names.append(arr)
24
  return names, probss
25
 
26
+ def create_solutions(image_urls, names, probss):
27
  solutions = [] #list to store all the objects
28
 
29
+ for image_url, class_name, prob, fileID in zip(image_urls, names, probss):
30
  obj = {"image": image_url, "answer": [class_name], "qcUser" : None}
31
  solutions.append(obj)
32
  print("image done")
 
49
  return {"error": f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}"}
50
 
51
  image_urls = params.get("urls", [])
52
+ # normalFileId = 0
53
  # normalFileId = params.get("normalfileID",[])
54
  # api = params.get("api", "")
55
  # job_id = params.get("job_id", "")
 
65
  return {"error": f"Error loading images: {str(e)}"}
66
 
67
  names, probss = detect_objects(images) # Perform object detection
68
+ solutions = create_solutions(image_urls, names, probss) # Create solutions with image URLs and bounding boxes
69
 
70
  # result_url = f"{api}/{job_id}"
71
  # send_results_to_api(solutions, result_url)