Hugging Face's logo

Spaces:
rxavier
/
off-topic-images
Runtime error

App Files Community
rxavier commited on
Commit
d0490bb
·
1 Parent(s): 4140edd

Update off_topic.py

Browse files
Files changed (1) hide show
  1. off_topic.py +2 -1
off_topic.py CHANGED
@@ -28,7 +28,8 @@ class OffTopicDetector:
28
  if valid_templates:
29
  valid_classes = [template.format(domain) for template in valid_templates]
30
  else:
31
- valid_classes = [f"a photo of {domain}", f"brochure with {domain} image", f"instructions for {domain}", f"{domain} diagram"]
 
32
  if not invalid_classes:
33
  invalid_classes = ["promotional ad with store information", "promotional text", "google maps screenshot", "business card", "qr code"]
34
  n_valid = len(valid_classes)
 
28
  if valid_templates:
29
  valid_classes = [template.format(domain) for template in valid_templates]
30
  else:
31
+ valid_classes = [f"a photo of {domain}", f"brochure with {domain} image", f"instructions for {domain}", f"{domain} diagram",
32
+ f"{domain} packaging", f"box of {domain}"]
33
  if not invalid_classes:
34
  invalid_classes = ["promotional ad with store information", "promotional text", "google maps screenshot", "business card", "qr code"]
35
  n_valid = len(valid_classes)