Upload 2 files

app.py

@@ -13,7 +13,7 @@ def convert_text_fre(input_text, grade_level, input_reading_score):
     output_text, similarity, reading_level, message = generate_sim_fre(input_text, min_level, max_level, MIN_ENTAILMENT, SYSTEM_PROMPT, MAX_ITER, float(input_reading_score))
     return output_text, similarity, reading_level, message
 
-def convert_text_sbert(input_text, grade_level, input_reading_score, is_fre):
+def convert_text_sbert(input_text, grade_level, input_reading_score):
     min_level, max_level = sbert_levels[grade_level]
     output_text, similarity, reading_level, message = generate_sim_sbert(input_text, min_level, max_level, MIN_ENTAILMENT, SYSTEM_PROMPT, MAX_ITER, float(input_reading_score))
     return output_text, similarity, reading_level, message
@@ -76,7 +76,7 @@ with gr.Blocks(css='styles.css') as app:
 
         convert_button.click(
             fn=convert_text_sbert,
-            inputs=[input_text, grade_level_fre, output_input_reading_score],
+            inputs=[input_text, grade_level_sbert, output_input_reading_score],
             outputs=[output_converted_text, output_similarity, output_reading_level, output_message]
         )
 

text_converter.py

@@ -62,10 +62,18 @@ def __set_reading_levels(level_type):
 
 def fetch_readability_score_fre(input_text):
     __set_reading_levels(__model_types[0])
+    print(f"=============== Setting Reading Levels to FRE ===============")
+    print(f"Reading Level Dictionary: {__reading_levels}\n")
+    print(f"Reading Level Dictionary: {__inverse_reading_levels}")
+    print(f"=============== Finished Setting Reading Levels to FRE ===============")
     return __user_input_readability_level(input_text, True)
 
 def fetch_readability_score_sbert(input_text):
     __set_reading_levels(__model_types[1])
+    print(f"=============== Setting Reading Levels to SBERT ===============")
+    print(f"Reading Level Dictionary: {__reading_levels}\n")
+    print(f"Reading Level Dictionary: {__inverse_reading_levels}")
+    print(f"=============== Finished Setting Reading Levels to SBERT ===============")
     return __user_input_readability_level(input_text, False)
 
 def __user_input_readability_level(input_text, is_fre):