Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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",
|