JuanMa360 commited on
Commit
fb7055a
·
1 Parent(s): 10b78eb

bugfix: living spaces function

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -159,8 +159,8 @@ def generate_answer(image):
159
  "view_type": {"views": None, "certainty_score": None}
160
  }
161
 
162
- max_space_type = max(space_type_score, key=space_type_score.get)
163
  space_type_score = calculate_space_type(processed_image)
 
164
  if space_type_score[max_space_type] >= threshold:
165
  space_type = max_space_type.lower()
166
  if space_type == "patio":
 
159
  "view_type": {"views": None, "certainty_score": None}
160
  }
161
 
 
162
  space_type_score = calculate_space_type(processed_image)
163
+ max_space_type = max(space_type_score, key=space_type_score.get)
164
  if space_type_score[max_space_type] >= threshold:
165
  space_type = max_space_type.lower()
166
  if space_type == "patio":