Update
Browse files
app.py
CHANGED
@@ -190,7 +190,7 @@ def create_app():
|
|
190 |
ex['license'],
|
191 |
]
|
192 |
for ex in all_examples
|
193 |
-
if ex['model']
|
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
|
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 |
**{
|