Spaces:
Sleeping
Sleeping
Commit
•
5b74304
1
Parent(s):
116ac08
Update text_converter.py
Browse filesAdding 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.
- 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 |
|