HakshaySundar commited on
Commit
5b74304
1 Parent(s): 116ac08

Update text_converter.py

Browse files

Adding a similarity check to ensure that the result not only has the lowest distance from target band but is also highly similar to the input text.

Files changed (1) hide show
  1. text_converter.py +3 -2
text_converter.py CHANGED
@@ -93,8 +93,9 @@ def generate_similar_sentence(input_text, min_reading_level, max_reading_level,
93
  result_index = i
94
  completed = True
95
  break
96
- elif (reading_level < min_reading_level and abs(reading_level - min_reading_level) < closeness) \
97
- or (reading_level > max_reading_level and abs(reading_level - max_reading_level) < closeness):
 
98
  closeness = abs(reading_level - min_reading_level) if reading_level < min_reading_level else abs(reading_level - max_reading_level)
99
  result_index = i
100
 
 
93
  result_index = i
94
  completed = True
95
  break
96
+ elif ((reading_level < min_reading_level and abs(reading_level - min_reading_level) < closeness) \
97
+ or (reading_level > max_reading_level and abs(reading_level - max_reading_level) < closeness)) and \
98
+ result_index > -1 and similarity > generated_text_scores[result_index][0]:
99
  closeness = abs(reading_level - min_reading_level) if reading_level < min_reading_level else abs(reading_level - max_reading_level)
100
  result_index = i
101