Update app_inference.py
Browse files- app_inference.py +2 -1
app_inference.py
CHANGED
@@ -13,6 +13,7 @@ from inf import InferencePipeline
|
|
13 |
|
14 |
|
15 |
SAMPLE_MODEL_IDS = [
|
|
|
16 |
'lora-library/B-LoRA-teddybear',
|
17 |
'lora-library/B-LoRA-bull',
|
18 |
'lora-library/B-LoRA-wolf_plushie',
|
@@ -103,7 +104,7 @@ css = """
|
|
103 |
def get_choices(hf_token):
|
104 |
api = HfApi(token=hf_token)
|
105 |
choices = [
|
106 |
-
info.modelId for info in api.list_models(author='
|
107 |
]
|
108 |
models_list = ['None'] + SAMPLE_MODEL_IDS + choices
|
109 |
return models_list
|
|
|
13 |
|
14 |
|
15 |
SAMPLE_MODEL_IDS = [
|
16 |
+
'thliang01/medieval-knight-sdxl-dora-v0-6',
|
17 |
'lora-library/B-LoRA-teddybear',
|
18 |
'lora-library/B-LoRA-bull',
|
19 |
'lora-library/B-LoRA-wolf_plushie',
|
|
|
104 |
def get_choices(hf_token):
|
105 |
api = HfApi(token=hf_token)
|
106 |
choices = [
|
107 |
+
info.modelId for info in api.list_models(author='thliang01')
|
108 |
]
|
109 |
models_list = ['None'] + SAMPLE_MODEL_IDS + choices
|
110 |
return models_list
|