skkk commited on
Commit
808f294
·
1 Parent(s): 5875991

add some logs

Browse files
Files changed (1) hide show
  1. Rodin.py +2 -0
Rodin.py CHANGED
@@ -131,10 +131,12 @@ class Generator:
131
 
132
  def preprocess(self, prompt, image_path, cache_image_base64, task_uuid=""):
133
  if cache_image_base64 and prompt and task_uuid != "":
 
134
  return prompt, cache_image_base64
135
  print("Preprocessing image...")
136
  success = False
137
  while not success:
 
138
  image_file = load_image(image_path)
139
  if prompt and task_uuid:
140
  preprocess_response = rodin_preprocess_image(generate_prompt=False, image=image_file, name="images.png", token=self.token)
 
131
 
132
  def preprocess(self, prompt, image_path, cache_image_base64, task_uuid=""):
133
  if cache_image_base64 and prompt and task_uuid != "":
134
+ print("Using cached image and prompt...")
135
  return prompt, cache_image_base64
136
  print("Preprocessing image...")
137
  success = False
138
  while not success:
139
+ print("Trying to preprocess image...")
140
  image_file = load_image(image_path)
141
  if prompt and task_uuid:
142
  preprocess_response = rodin_preprocess_image(generate_prompt=False, image=image_file, name="images.png", token=self.token)