Update app_6.py

app_6.py

@@ -138,7 +138,7 @@ def extract_arguments(text, tokenizer, model, beam_search=True):
         # Inserts tags into the original text based on token offsets.
         if start_token is not None and end_token is not None:
 
-            st.write(f"Start: {start_token}, End: {end_token}")
+            #st.write(f"Start: {start_token}, End: {end_token}")
             
 
             if start_token > end_token:
@@ -150,7 +150,7 @@ def extract_arguments(text, tokenizer, model, beam_search=True):
                 start_idx, end_idx = offset_mapping[start_token][0], offset_mapping[end_token][1]
                  
                 if start_idx is not None and end_idx is not None and start_idx < end_idx:
-                    st.write(f"Start_idx: {start_idx}, End_idx: {end_idx}")
+                    #st.write(f"Start_idx: {start_idx}, End_idx: {end_idx}")
                     
                     return (
                         original_text[:start_idx]