JagmeetMinhas22 commited on
Commit
a1bd472
·
verified ·
1 Parent(s): 5164b35

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -4,15 +4,14 @@ import gradio as gr
4
  model = gr.load("models/stabilityai/stable-diffusion-3.5-large")
5
 
6
  # Define a function to handle the model inference
7
- def generate_image(prompt, num_inference_steps):
8
- return model(prompt=prompt, num_inference_steps=num_inference_steps)
9
 
10
  # Define the Gradio interface
11
  interface = gr.Interface(
12
  fn=generate_image, # The prediction function
13
  inputs=[
14
  gr.Textbox(label="Prompt"), # Input for the text prompt
15
- gr.Slider(minimum=1, maximum=100, label="Number of Inference Steps") # Slider for steps
16
  ],
17
  outputs=gr.Image(label="Generated Image") # Output component for the generated image
18
  )
 
4
  model = gr.load("models/stabilityai/stable-diffusion-3.5-large")
5
 
6
  # Define a function to handle the model inference
7
+ def generate_image(prompt):
8
+ return model(prompt=prompt)
9
 
10
  # Define the Gradio interface
11
  interface = gr.Interface(
12
  fn=generate_image, # The prediction function
13
  inputs=[
14
  gr.Textbox(label="Prompt"), # Input for the text prompt
 
15
  ],
16
  outputs=gr.Image(label="Generated Image") # Output component for the generated image
17
  )