Spaces:
Runtime error
Runtime error
Update off_topic.py
Browse files- 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)
|