Spaces:

JuanMa360
/

vision_intelligence

Sleeping

JuanMa360 commited on May 30, 2024

Commit

cff17c0

1 Parent(s): 991ef04

refactor: pool

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ processor = CLIPProcessor.from_pretrained("tokenizer")
 vqa_pipeline = pipeline("visual-question-answering")
 space_type_labels = ["living room", "bedroom", "kitchen", "terrace", "closet","bathroom", "dining room", "office", "garage", "garden",
-    "balcony", "attic", "hallway", "laundry room","gym", "playroom", "storage room", "studio","is_exterior","empty_interior_room","others"]
 equipment_questions = [
     "Does the image show outdoor furniture?",

 vqa_pipeline = pipeline("visual-question-answering")
 space_type_labels = ["living room", "bedroom", "kitchen", "terrace", "closet","bathroom", "dining room", "office", "garage", "garden",
+    "balcony", "attic", "hallway", "laundry room","gym", "playroom", "storage room", "studio","is_exterior","empty_interior_room","swimming pool"]
 equipment_questions = [
     "Does the image show outdoor furniture?",