skkk commited on
Commit
b27cd85
·
1 Parent(s): 37679c0

add some warning

Browse files
Files changed (1) hide show
  1. Rodin.py +4 -4
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.png", token=self.token)
155
  else:
156
- preprocess_response = rodin_preprocess_image(generate_prompt=True, image=image_file, name="images.png", 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,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 e
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