Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,17 +10,17 @@ pipes = {
|
|
10 |
"ViT/L-14": pipeline("zero-shot-image-classification", model="openai/clip-vit-base-patch16"),
|
11 |
}
|
12 |
inputs = [
|
13 |
-
gr.
|
14 |
label="Image"),
|
15 |
-
gr.
|
16 |
label="Candidate Labels"),
|
17 |
-
gr.
|
18 |
"ViT/B-16",
|
19 |
"ViT/L-14",
|
20 |
"ViT/L-14@336px",
|
21 |
"ViT/H-14",
|
22 |
], type="value", default="ViT/B-16", label="Model"),
|
23 |
-
gr.
|
24 |
label="Prompt Template Prompt",
|
25 |
default="a photo of a {}"),
|
26 |
]
|
|
|
10 |
"ViT/L-14": pipeline("zero-shot-image-classification", model="openai/clip-vit-base-patch16"),
|
11 |
}
|
12 |
inputs = [
|
13 |
+
gr.Image(type='pil',
|
14 |
label="Image"),
|
15 |
+
gr.Textbox(lines=1,
|
16 |
label="Candidate Labels"),
|
17 |
+
gr.Radio(choices=[
|
18 |
"ViT/B-16",
|
19 |
"ViT/L-14",
|
20 |
"ViT/L-14@336px",
|
21 |
"ViT/H-14",
|
22 |
], type="value", default="ViT/B-16", label="Model"),
|
23 |
+
gr.Textbox(lines=1,
|
24 |
label="Prompt Template Prompt",
|
25 |
default="a photo of a {}"),
|
26 |
]
|