Spaces:
Build error
Build error
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 |
|