Spaces:
Runtime error
Runtime error
Martijn van Beers
commited on
Commit
·
1b49495
1
Parent(s):
f992b4c
Change defult model
Browse files
app.py
CHANGED
@@ -58,7 +58,7 @@ def run_demo(*args):
|
|
58 |
image, text, model_name = args
|
59 |
elif len(args) == 2:
|
60 |
image, text = args
|
61 |
-
model_name = "ViT-
|
62 |
else:
|
63 |
raise ValueError("Unexpected number of parameters")
|
64 |
|
@@ -86,7 +86,7 @@ def run_demo(*args):
|
|
86 |
default_inputs = [
|
87 |
gr.components.Image(type='pil', label="Original Image"),
|
88 |
gr.components.Textbox(label="Image description"),
|
89 |
-
gr.Dropdown(label="CLIP Model", choices=['ViT-B/16', 'ViT-B/32', 'ViT-L/14'], value="ViT-
|
90 |
]
|
91 |
|
92 |
default_outputs = [
|
|
|
58 |
image, text, model_name = args
|
59 |
elif len(args) == 2:
|
60 |
image, text = args
|
61 |
+
model_name = "ViT-B/32"
|
62 |
else:
|
63 |
raise ValueError("Unexpected number of parameters")
|
64 |
|
|
|
86 |
default_inputs = [
|
87 |
gr.components.Image(type='pil', label="Original Image"),
|
88 |
gr.components.Textbox(label="Image description"),
|
89 |
+
gr.Dropdown(label="CLIP Model", choices=['ViT-B/16', 'ViT-B/32', 'ViT-L/14'], value="ViT-B/32"),
|
90 |
]
|
91 |
|
92 |
default_outputs = [
|