xnetba commited on
Commit
acf0fde
·
1 Parent(s): ddec46e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ default_model = "stabilityai/stable-diffusion-2"
23
  model_name = gr.inputs.Dropdown(choices=model_list, label="Select Model", default=default_model)
24
 
25
  def generate_image(text, model_name):
26
- model = gradio.interface.load_model(model_name, source="huggingface", api_key=api_key)
27
  return model.predict(text)
28
 
29
  input_text = gr.inputs.Textbox(label="Input Text")
 
23
  model_name = gr.inputs.Dropdown(choices=model_list, label="Select Model", default=default_model)
24
 
25
  def generate_image(text, model_name):
26
+ model = gr.interface.load_model(model_name, source="huggingface", api_key=api_key)
27
  return model.predict(text)
28
 
29
  input_text = gr.inputs.Textbox(label="Input Text")