DigiP-AI commited on
Commit
ffbcee1
·
verified ·
1 Parent(s): 808a6f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -289,10 +289,10 @@ with gr.Blocks(theme=theme, css=css) as app:
289
  #gr.load("models/XLabs-AI/flux-RealismLora")
290
  model="./models/XLabs-AI/flux-RealismLora"
291
  def model(image):
292
- return model(image)[0][Image]
293
  text_prompt = gr.Textbox(label="Prompt", placeholder="Enter a prompt")
294
  image_output = gr.Image(type="pil", label="Image Output", format="png"
295
- text_btn = gr.Button(Run)
296
  text_btn.click(model, text_prompt, image_output)
297
 
298
 
 
289
  #gr.load("models/XLabs-AI/flux-RealismLora")
290
  model="./models/XLabs-AI/flux-RealismLora"
291
  def model(image):
292
+ return model(image)[0]["Image"]
293
  text_prompt = gr.Textbox(label="Prompt", placeholder="Enter a prompt")
294
  image_output = gr.Image(type="pil", label="Image Output", format="png"
295
+ text_btn = gr.Button("Run")
296
  text_btn.click(model, text_prompt, image_output)
297
 
298