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
|