JuanMa360 commited on
Commit
e40ee10
·
1 Parent(s): c391954

bugfix: views

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -179,7 +179,7 @@ def generate_answer(image):
179
  luminosity_score = calculate_luminosity_score(processed_image)
180
  image_data["luminosity_score"]['score'] = luminosity_score['natural_light']
181
 
182
- result = vqa_pipeline(processed_image, "Is there a real window frame?", top_k=1)
183
  has_window = result[0]
184
  image_data["luminosity_score"]["has_window"] = has_window
185
 
@@ -190,8 +190,8 @@ def generate_answer(image):
190
  certainty_score = list(certainty_score.values())[0]
191
  image_data["view_type"]["certainty_score"] = certainty_score
192
 
193
- is_render = calculate_is_render(processed_image)
194
- image_data["is_render"] = is_render
195
 
196
  return json.dumps(image_data, indent=4)
197
 
 
179
  luminosity_score = calculate_luminosity_score(processed_image)
180
  image_data["luminosity_score"]['score'] = luminosity_score['natural_light']
181
 
182
+ result = vqa_pipeline(processed_image, "Is there a real window frame or picture window?", top_k=1)
183
  has_window = result[0]
184
  image_data["luminosity_score"]["has_window"] = has_window
185
 
 
190
  certainty_score = list(certainty_score.values())[0]
191
  image_data["view_type"]["certainty_score"] = certainty_score
192
 
193
+ #is_render = calculate_is_render(processed_image)
194
+ #image_data["is_render"] = is_render
195
 
196
  return json.dumps(image_data, indent=4)
197