Init files

README.md

@@ -1,13 +1,13 @@
 ---
-title: CzechPunctuation
-emoji: 🌖
-colorFrom: blue
-colorTo: pink
+title: CzechCapitalization
+emoji: 😻
+colorFrom: green
+colorTo: purple
 sdk: streamlit
-sdk_version: 1.17.0
+sdk_version: 1.21.0
 app_file: app.py
 pinned: false
-license: gpl-3.0
+license: cc-by-nc-4.0
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,29 @@
+import streamlit as st
+from huggingface_hub import Repository
+
+repo = Repository(
+    local_dir="scripts",
+    repo_type="model",
+    clone_from="stistko/CzechPunctuationProtected",
+    token=True
+)
+repo.git_pull()
+
+from scripts.scripts.model import PunctuationModel
+
+model = PunctuationModel()
+
+st.write("""
+# Czech Punctuation Model (CPM)
+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:")
+
+# Button for submission
+submit_button = st.button("Submit")
+
+# Action after button press
+if submit_button:
+    st.write(f"Output: {model.run(input_text)}")

requirements.txt

@@ -0,0 +1,7 @@
+torch
+tqdm
+numpy
+sentencepiece
+transformers
+scikit-learn
+huggingface_hub