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 |
|