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
|