Update app.py

app.py

@@ -5,9 +5,7 @@ from transformers import T5ForConditionalGeneration, T5TokenizerFast, T5Config
 st.session_state
 if 'textbox' not in st.session_state:
     st.session_state['textbox'] = "Vi bruker ikke tegnsetting eller store bokstaver når vi prater. Vi slår også sammen ord, og i praksis er dermed heller ikke mellomrom meningsbærende. Prøv å fjerne tegnsetting, store bokstaver og mellomrom fra dette avsnittet. Se om den nye North-T5-modellen greier å sette sammen til et nytt meningsbærende avsnitt."
-if 'selectbox' not in st.session_state:
-    st.session_state['selectbox'] = ""
-    
+ 
 
 
 @st.cache(allow_output_mutation=True, suppress_st_warning=True)
@@ -27,6 +25,15 @@ def deuncase(model, tokenizer, text):
 def uncase():
     st.session_state['textbox'] = st.session_state['textbox'].lower()
 
+def unpunct():
+    text = st.session_state['textbox']
+    trans_chars = "'\",.:;-_*?/\n"
+    trans_table = text.maketrans("", "", trans_chars)
+    st.session_state['textbox'] = text.translate(trans_table)
+    
+def unspace():
+    st.session_state['textbox'] = st.session_state['textbox'].replace(" ","")
+    
 def sidebar_callback():
     st.session_state['textbox'] = st.session_state['selectbox']