xnetba commited on
Commit
1cd1277
·
1 Parent(s): 3509ad8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,12 +20,12 @@ model_list = [
20
  ]
21
 
22
  default_model = "stabilityai/stable-diffusion-2"
23
- model_name = gr.inputs.Dropdown(model_list, label="Select Model", default=default_model)
24
  model = None
25
 
26
  def load_model(model_name):
27
  global model
28
- model = gr.load(model_name, source="huggingface", api_key=api_key)
29
 
30
  def predict(inputs):
31
  return model.predict(inputs)
 
20
  ]
21
 
22
  default_model = "stabilityai/stable-diffusion-2"
23
+ model_name = gr.inputs.Dropdown(choices=model_list, label="Select Model", default=default_model)
24
  model = None
25
 
26
  def load_model(model_name):
27
  global model
28
+ model = gr.Interface.load(model_name, source="huggingface", api_key=api_key)
29
 
30
  def predict(inputs):
31
  return model.predict(inputs)