Shuaizhang7 commited on
Commit
04ac035
·
verified ·
1 Parent(s): 50f82bd

Update app.py

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