Aluren commited on
Commit
7286300
·
verified ·
1 Parent(s): cea1e2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -6,7 +6,7 @@ from typing import Any, List, Union
6
  import gradio as gr
7
  import numpy as np
8
  import torch
9
- from gradio_image_prompter import ImagePrompter
10
  from gradio_litmodel3d import LitModel3D
11
  from huggingface_hub import snapshot_download
12
  from PIL import Image
@@ -157,7 +157,7 @@ def run_refinement(
157
  if randomize_seed:
158
  seed = random.randint(0, MAX_SEED)
159
 
160
- # print("rgb_image", rgb_image)
161
  # print("mesh", rgb_image)
162
 
163
  if not isinstance(rgb_image, Image.Image) and "image" in rgb_image:
@@ -186,7 +186,8 @@ with gr.Blocks() as demo:
186
  with gr.Row():
187
  with gr.Column():
188
  with gr.Row():
189
- image_prompts = ImagePrompter(label="Input Image", type="pil")
 
190
  mesh = gr.Model3D(label="Input Coarse Model",camera_position=(90,90,3))
191
 
192
  with gr.Accordion("Generation Settings", open=False):
 
6
  import gradio as gr
7
  import numpy as np
8
  import torch
9
+ # from gradio_image_prompter import ImagePrompter
10
  from gradio_litmodel3d import LitModel3D
11
  from huggingface_hub import snapshot_download
12
  from PIL import Image
 
157
  if randomize_seed:
158
  seed = random.randint(0, MAX_SEED)
159
 
160
+ print("rgb_image", rgb_image)
161
  # print("mesh", rgb_image)
162
 
163
  if not isinstance(rgb_image, Image.Image) and "image" in rgb_image:
 
186
  with gr.Row():
187
  with gr.Column():
188
  with gr.Row():
189
+ # image_prompts = ImagePrompter(label="Input Image", type="pil")
190
+ image_prompts = gr.Image(label="Example Image", type="filepath", interactive=False, visible=False)
191
  mesh = gr.Model3D(label="Input Coarse Model",camera_position=(90,90,3))
192
 
193
  with gr.Accordion("Generation Settings", open=False):