Spaces:
Runtime error
Runtime error
Shuaizhang7
commited on
Commit
•
de0bb86
1
Parent(s):
b12da12
Update app.py
Browse files
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",
|