dezzman commited on
Commit
cec3a8b
·
verified ·
1 Parent(s): e5f2339

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -53,7 +53,7 @@ def infer(
53
  else:
54
  image = pipe_default(**params).images[0]
55
 
56
- return image, pipe.name_or_path
57
 
58
  css = """
59
  #col-container {
@@ -135,7 +135,6 @@ with gr.Blocks(css=css) as demo:
135
 
136
  run_button = gr.Button("Run", scale=1, variant="primary")
137
  result = gr.Image(label="Result", show_label=False)
138
- inferenced_model_name = gr.Label(show_label=True, value=model_id.value)
139
 
140
  gr.on(
141
  triggers=[run_button.click, prompt.submit],
@@ -150,10 +149,7 @@ with gr.Blocks(css=css) as demo:
150
  seed,
151
  guidance_scale,
152
  ],
153
- outputs=[
154
- result,
155
- inferenced_model_name,
156
- ],
157
  )
158
 
159
  if __name__ == "__main__":
 
53
  else:
54
  image = pipe_default(**params).images[0]
55
 
56
+ return image
57
 
58
  css = """
59
  #col-container {
 
135
 
136
  run_button = gr.Button("Run", scale=1, variant="primary")
137
  result = gr.Image(label="Result", show_label=False)
 
138
 
139
  gr.on(
140
  triggers=[run_button.click, prompt.submit],
 
149
  seed,
150
  guidance_scale,
151
  ],
152
+ outputs=[result],
 
 
 
153
  )
154
 
155
  if __name__ == "__main__":