Martijn van Beers commited on
Commit
1b49495
1 Parent(s): f992b4c

Change defult model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-L/14"
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-L/14"),
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 = [