quentin99 commited on
Commit
7fb76ee
1 Parent(s): 49cdb29

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import numpy as np
5
  from PIL import Image
6
 
7
  pipes = {
8
- "ViT/B-16": pipeline("zero-shot-test", model="OFA-Sys/chinese-clip-vit-base-patch16")
9
  }
10
 
11
  inputs = [
 
5
  from PIL import Image
6
 
7
  pipes = {
8
+ "ViT/B-16": pipeline("zero-shot-image-classification", model="OFA-Sys/chinese-clip-vit-base-patch16")
9
  }
10
 
11
  inputs = [