skkk commited on
Commit
c15f9cf
1 Parent(s): b27cd85

Modified some tips

Browse files
Files changed (3) hide show
  1. .DS_Store +0 -0
  2. Rodin.py +2 -2
  3. app.py +1 -1
.DS_Store CHANGED
Binary files a/.DS_Store and b/.DS_Store differ
 
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.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']}")
 
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=os.path.basename(image_path), token=self.token)
155
  else:
156
+ preprocess_response = rodin_preprocess_image(generate_prompt=True, image=image_file, name=os.path.basename(image_path), 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']}")
app.py CHANGED
@@ -159,7 +159,7 @@ with gr.Blocks() as demo:
159
  block_contrast = gr.Button("Contrast", min_width=0)
160
  block_normal = gr.Button("Normal", min_width=0)
161
 
162
- button_more = gr.Button(value="Download", variant="primary", link=rodin_url)
163
  gr.Markdown("""
164
  **HELP**:
165
  1. Upload an image to generate 3D geometry.
 
159
  block_contrast = gr.Button("Contrast", min_width=0)
160
  block_normal = gr.Button("Normal", min_width=0)
161
 
162
+ button_more = gr.Button(value="Download from Rodin", variant="primary", link=rodin_url)
163
  gr.Markdown("""
164
  **HELP**:
165
  1. Upload an image to generate 3D geometry.