thliang01 commited on
Commit
c5ddec7
·
verified ·
1 Parent(s): b87e971

Update app_inference.py

Browse files
Files changed (1) hide show
  1. 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='lora-library')
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