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.json()
|
| 56 |
|
| 57 |
def crop_image(image, type):
|
| 58 |
if image == None:
|
|
@@ -154,11 +154,17 @@ class Generator:
|
|
| 154 |
try_times += 1
|
| 155 |
image_file = load_image(image_path)
|
| 156 |
log("INFO", "Image loaded, processing...")
|
| 157 |
-
|
| 158 |
-
|
| 159 |
-
|
| 160 |
-
|
| 161 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 162 |
if 'error' in preprocess_response:
|
| 163 |
log("ERROR", f"Error in image preprocessing: {preprocess_response['error']}")
|
| 164 |
raise RuntimeError
|
|
|
|
| 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:
|
|
|
|
| 154 |
try_times += 1
|
| 155 |
image_file = load_image(image_path)
|
| 156 |
log("INFO", "Image loaded, processing...")
|
| 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}")
|
| 166 |
+
raise gr.Error("Error in image preprocessing, please try again.")
|
| 167 |
+
|
| 168 |
if 'error' in preprocess_response:
|
| 169 |
log("ERROR", f"Error in image preprocessing: {preprocess_response['error']}")
|
| 170 |
raise RuntimeError
|