Shuaizhang7 commited on
Commit
de0bb86
1 Parent(s): b12da12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -6,6 +6,7 @@ from PIL import Image
6
 
7
 
8
  pipes = {
 
9
  "ViT/B-16": pipeline("zero-shot-image-classification", model="OFA-Sys/chinese-clip-vit-base-patch16"),
10
  "ViT/L-14": pipeline("zero-shot-image-classification", model="OFA-Sys/chinese-clip-vit-large-patch14"),
11
  "ViT/L-14@336px": pipeline("zero-shot-image-classification", model="OFA-Sys/chinese-clip-vit-large-patch14-336px"),
@@ -16,7 +17,8 @@ inputs = [
16
  label="Image 输入图片"),
17
  gr.inputs.Textbox(lines=1,
18
  label="Candidate Labels 候选分类标签"),
19
- gr.inputs.Radio(choices=[
 
20
  "ViT/B-16",
21
  "ViT/L-14",
22
  "ViT/L-14@336px",
 
6
 
7
 
8
  pipes = {
9
+ "openAI-ViT/B-16": pipeline("zero-shot-image-classification", model="openai/clip-vit-base-patch16"),
10
  "ViT/B-16": pipeline("zero-shot-image-classification", model="OFA-Sys/chinese-clip-vit-base-patch16"),
11
  "ViT/L-14": pipeline("zero-shot-image-classification", model="OFA-Sys/chinese-clip-vit-large-patch14"),
12
  "ViT/L-14@336px": pipeline("zero-shot-image-classification", model="OFA-Sys/chinese-clip-vit-large-patch14-336px"),
 
17
  label="Image 输入图片"),
18
  gr.inputs.Textbox(lines=1,
19
  label="Candidate Labels 候选分类标签"),
20
+ gr.inputs.Radio(choices=[
21
+ "openAI-ViT/B-16"
22
  "ViT/B-16",
23
  "ViT/L-14",
24
  "ViT/L-14@336px",