skkk commited on
Commit
d2ad507
·
1 Parent(s): 8db01c3
Files changed (1) hide show
  1. Rodin.py +4 -3
Rodin.py CHANGED
@@ -52,7 +52,7 @@ def rodin_preprocess_image(generate_prompt, image, name, token):
52
  'Authorization': f'Bearer {token}'
53
  }
54
  response = requests.post(f"{BASE_URL}/task/rodin_mesh_image_process", data=m, headers=headers)
55
- return response, response.json()
56
 
57
  def crop_image(image, type):
58
  if image == None:
@@ -157,9 +157,10 @@ class Generator:
157
 
158
  try:
159
  if prompt and task_uuid:
160
- res, preprocess_response = rodin_preprocess_image(generate_prompt=False, image=image_file, name=os.path.basename(image_path), token=self.token)
161
  else:
162
- res, preprocess_response = rodin_preprocess_image(generate_prompt=True, image=image_file, name=os.path.basename(image_path), token=self.token)
 
163
  log("INFO", f"Image preprocessed: {preprocess_response.get('statusCode')}")
164
  except Exception as e:
165
  log("ERROR", f"Error in image preprocessing: {res}")
 
52
  'Authorization': f'Bearer {token}'
53
  }
54
  response = requests.post(f"{BASE_URL}/task/rodin_mesh_image_process", data=m, headers=headers)
55
+ return response
56
 
57
  def crop_image(image, type):
58
  if image == None:
 
157
 
158
  try:
159
  if prompt and task_uuid:
160
+ res = rodin_preprocess_image(generate_prompt=False, image=image_file, name=os.path.basename(image_path), token=self.token)
161
  else:
162
+ res = rodin_preprocess_image(generate_prompt=True, image=image_file, name=os.path.basename(image_path), token=self.token)
163
+ preprocess_response = res.json()
164
  log("INFO", f"Image preprocessed: {preprocess_response.get('statusCode')}")
165
  except Exception as e:
166
  log("ERROR", f"Error in image preprocessing: {res}")