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}")
|