yangapku commited on
Commit
a06e4eb
1 Parent(s): 859f190

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -17,8 +17,9 @@ def shot(image, labels_text):
17
  PIL_image = Image.fromarray(np.uint8(image)).convert('RGB')
18
  labels = labels_text.split(",")
19
  res = pipes['chinese-clip-vit-base-patch16'](images=PIL_image,
20
- candidate_labels=labels,
21
- hypothesis_template= "This example is {}.")
 
22
  return {dic["label"]: dic["score"] for dic in res}
23
 
24
  iface = gr.Interface(shot,
 
17
  PIL_image = Image.fromarray(np.uint8(image)).convert('RGB')
18
  labels = labels_text.split(",")
19
  res = pipes['chinese-clip-vit-base-patch16'](images=PIL_image,
20
+ # candidate_labels=labels,
21
+ # hypothesis_template= "This example is {}.")
22
+ candidate_labels=labels)
23
  return {dic["label"]: dic["score"] for dic in res}
24
 
25
  iface = gr.Interface(shot,