Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|