D3V1L1810 commited on
Commit
bbfb731
·
verified ·
1 Parent(s): c6dbf09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -47,7 +47,10 @@ def process_images(params):
47
  return {"error": f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}"}
48
 
49
  image_urls = params.get("urls", [])
50
- file_id = params.get("normalfileID",[])
 
 
 
51
  # api = params.get("api", "")
52
  # job_id = params.get("job_id", "")
53
 
@@ -62,7 +65,7 @@ def process_images(params):
62
  return {"error": f"Error loading images: {str(e)}"}
63
 
64
  names = detect_objects(images)
65
- solutions = create_solutions(image_urls, names,file_id)
66
 
67
  # result_url = f"{api}/{job_id}"
68
  # response = send_results_to_api(solutions, result_url)
 
47
  return {"error": f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}"}
48
 
49
  image_urls = params.get("urls", [])
50
+ if not params.get("normalfileID",[]):
51
+ file_ids = [None]*len(image_urls)
52
+ else:
53
+ file_ids = params.get("normalfileID",[])
54
  # api = params.get("api", "")
55
  # job_id = params.get("job_id", "")
56
 
 
65
  return {"error": f"Error loading images: {str(e)}"}
66
 
67
  names = detect_objects(images)
68
+ solutions = create_solutions(image_urls, names, file_ids)
69
 
70
  # result_url = f"{api}/{job_id}"
71
  # response = send_results_to_api(solutions, result_url)