abetlen commited on
Commit
1ab8bf1
1 Parent(s): 0ce431b
Files changed (2) hide show
  1. app.py +1 -1
  2. models.py +1 -1
app.py CHANGED
@@ -190,7 +190,7 @@ def create_app():
190
  ex['license'],
191
  ]
192
  for ex in all_examples
193
- if ex['model'] + "-gguf" in models.MODELS
194
  ],
195
  inputs=[example_image, example_prompt, example_model, example_license],
196
  )
 
190
  ex['license'],
191
  ]
192
  for ex in all_examples
193
+ if ex['model'] in models.MODELS
194
  ],
195
  inputs=[example_image, example_prompt, example_model, example_license],
196
  )
models.py CHANGED
@@ -12,7 +12,7 @@ import gradio_helpers
12
 
13
  ORGANIZATION = 'abetlen'
14
  BASE_MODELS = [
15
- ('paligemma-3b-mix-224-gguf', 'paligemma-3b-mix-224-text-model-q4_k_m.gguf'),
16
  ]
17
  MODELS = {
18
  **{
 
12
 
13
  ORGANIZATION = 'abetlen'
14
  BASE_MODELS = [
15
+ ('paligemma-3b-mix-224-gguf', 'paligemma-3b-mix-224'),
16
  ]
17
  MODELS = {
18
  **{