Spaces:
Runtime error
Runtime error
fix a bug
Browse files
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
|
| 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
|
| 161 |
else:
|
| 162 |
-
res
|
|
|
|
| 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}")
|