Hugging Face's logo

Spaces:
Shuaizhang7
/
Surveillance_scene_recognition
Runtime error

App Files Community
Shuaizhang7 commited on
Commit
50f82bd
•
1 Parent(s): 8c7401e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -6,11 +6,11 @@ from PIL import Image
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"),
13
- "ViT/H-14": pipeline("zero-shot-image-classification", model="OFA-Sys/chinese-clip-vit-huge-patch14"),
14
  }
15
  inputs = [
16
  gr.inputs.Image(type='pil',
 
6
 
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"),
13
+ "CN-ViT/H-14": pipeline("zero-shot-image-classification", model="OFA-Sys/chinese-clip-vit-huge-patch14"),
14
  }
15
  inputs = [
16
  gr.inputs.Image(type='pil',