xnetba commited on
Commit
6495e28
·
1 Parent(s): 451adba

Update app.py

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