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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,9 +14,9 @@ pipes = {
14
  images="festival.jpg"
15
 
16
  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=image,
20
  candidate_labels=labels,
21
  hypothesis_template= "This example is {}.")
22
  return {dic["label"]: dic["score"] for dic in res}
 
14
  images="festival.jpg"
15
 
16
  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}