Update app.py

app.py

@@ -15,14 +15,17 @@ model = CapitalizationModel()
 
 # Inicializace
 
+st.write("""
+# Capitalization Model
+This application uses a transformer model 'Electra-small' with subsequent fine-tuning and a contextual window of 256 tokens.
+""")
 
+# Text input field
+input_text = st.text_input("Enter text here:")
 
-# Textové vstupní pole
-input_text = st.text_input("Zadejte text zde:")
+# Button for submission
+submit_button = st.button("Submit")
 
-# Tlačítko pro potvrzení
-submit_button = st.button("Odeslat")
-
-# Akce po stisknutí tlačítka
+# Action after button press
 if submit_button:
-    st.write(f"Výstup: {model.run(input_text)}")
+    st.write(f"Output: {model.run(input_text)}")