MaxMilan1 commited on
Commit
657d017
1 Parent(s): 8d4c07e
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -36,6 +36,10 @@ with gr.Blocks() as demo:
36
  with gr.Column():
37
  gen_image = gr.Image(label="Generated Image", show_download_button=True, show_label=False)
38
 
 
 
 
 
39
  with gr.Tab("Image to 3D Model Generator"):
40
  with gr.Row(variant="panel"):
41
  with gr.Column():
@@ -98,7 +102,6 @@ with gr.Blocks() as demo:
98
  gr.Markdown(_CITE_)
99
 
100
  mv_images = gr.State()
101
- gen_image = gr.State()
102
 
103
  submit.click(fn=check_input_image, inputs=[gen_image]).success(
104
  fn=preprocess,
@@ -114,6 +117,4 @@ with gr.Blocks() as demo:
114
  outputs=[output_model_obj]
115
  )
116
 
117
- button_gen.click(check_prompt, inputs=[prompt]).success(generate_image, inputs=[prompt, negative_prompt, control_image, scale], outputs=[gen_image])
118
-
119
  demo.launch()
 
36
  with gr.Column():
37
  gen_image = gr.Image(label="Generated Image", show_download_button=True, show_label=False)
38
 
39
+ gen_image = gr.State()
40
+
41
+ button_gen.click(check_prompt, inputs=[prompt]).success(generate_image, inputs=[prompt, negative_prompt, control_image, scale], outputs=[gen_image])
42
+
43
  with gr.Tab("Image to 3D Model Generator"):
44
  with gr.Row(variant="panel"):
45
  with gr.Column():
 
102
  gr.Markdown(_CITE_)
103
 
104
  mv_images = gr.State()
 
105
 
106
  submit.click(fn=check_input_image, inputs=[gen_image]).success(
107
  fn=preprocess,
 
117
  outputs=[output_model_obj]
118
  )
119
 
 
 
120
  demo.launch()