Spaces:
Runtime error
Runtime error
add some warning
Browse files
Rodin.py
CHANGED
|
@@ -151,9 +151,9 @@ class Generator:
|
|
| 151 |
image_file = load_image(image_path)
|
| 152 |
log("INFO", "Image loaded, processing...")
|
| 153 |
if prompt and task_uuid:
|
| 154 |
-
preprocess_response = rodin_preprocess_image(generate_prompt=False, image=image_file, name="images.
|
| 155 |
else:
|
| 156 |
-
preprocess_response = rodin_preprocess_image(generate_prompt=True, image=image_file, name="images.
|
| 157 |
log("INFO", f"Image preprocessed: {preprocess_response.get('statusCode')}")
|
| 158 |
if 'error' in preprocess_response:
|
| 159 |
log("ERROR", f"Error in image preprocessing: {preprocess_response['error']}")
|
|
@@ -165,12 +165,12 @@ class Generator:
|
|
| 165 |
else:
|
| 166 |
try:
|
| 167 |
if not (prompt and task_uuid):
|
| 168 |
-
prompt = preprocess_response.get('prompt', None)
|
| 169 |
processed_image = "data:image/png;base64," + preprocess_response.get('processed_image', None)
|
| 170 |
success = True
|
| 171 |
except Exception as e:
|
| 172 |
log("ERROR", f"Error in image preprocessing: {preprocess_response}")
|
| 173 |
-
raise
|
| 174 |
|
| 175 |
return prompt, processed_image
|
| 176 |
|
|
|
|
| 151 |
image_file = load_image(image_path)
|
| 152 |
log("INFO", "Image loaded, processing...")
|
| 153 |
if prompt and task_uuid:
|
| 154 |
+
preprocess_response = rodin_preprocess_image(generate_prompt=False, image=image_file, name="images.jpeg", token=self.token)
|
| 155 |
else:
|
| 156 |
+
preprocess_response = rodin_preprocess_image(generate_prompt=True, image=image_file, name="images.jpeg", token=self.token)
|
| 157 |
log("INFO", f"Image preprocessed: {preprocess_response.get('statusCode')}")
|
| 158 |
if 'error' in preprocess_response:
|
| 159 |
log("ERROR", f"Error in image preprocessing: {preprocess_response['error']}")
|
|
|
|
| 165 |
else:
|
| 166 |
try:
|
| 167 |
if not (prompt and task_uuid):
|
| 168 |
+
prompt = preprocess_response.get('prompt', None)
|
| 169 |
processed_image = "data:image/png;base64," + preprocess_response.get('processed_image', None)
|
| 170 |
success = True
|
| 171 |
except Exception as e:
|
| 172 |
log("ERROR", f"Error in image preprocessing: {preprocess_response}")
|
| 173 |
+
raise gr.Error("Busy connection, please try again later.")
|
| 174 |
|
| 175 |
return prompt, processed_image
|
| 176 |
|