FengHou97 commited on
Commit
0dd7d52
·
verified ·
1 Parent(s): e43af19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -36,8 +36,7 @@ images="festival.jpg"
36
  def shot(image, labels_text, model_name, hypothesis_template_prefix, hypothesis_template_suffix, domains_text):
37
  labels = [label.strip(" ") for label in labels_text.strip(" ").split(",")]
38
  domains = [domain.strip(" ") for domain in domains_text.strip(" ").split(",")]
39
- hypothesis_template_suffix.format(*domains)
40
- hypothesis_template = hypothesis_template_prefix + ' ' + hypothesis_template_suffix
41
  print(hypothesis_template)
42
 
43
  res = pipes[model_name](images=image,
 
36
  def shot(image, labels_text, model_name, hypothesis_template_prefix, hypothesis_template_suffix, domains_text):
37
  labels = [label.strip(" ") for label in labels_text.strip(" ").split(",")]
38
  domains = [domain.strip(" ") for domain in domains_text.strip(" ").split(",")]
39
+ hypothesis_template = hypothesis_template_prefix + ' ' + hypothesis_template_suffix.format(*domains)
 
40
  print(hypothesis_template)
41
 
42
  res = pipes[model_name](images=image,