Init

app.py

@@ -0,0 +1,28 @@
+import streamlit as st
+from huggingface_hub import Repository
+
+repo = Repository(
+    local_dir="scripts",
+    repo_type="model",
+    clone_from="stistko/CzechCapitalizationProtected",
+    token=True
+)
+repo.git_pull()
+
+from scripts.scripts.model import CapitalizationModel
+
+model = CapitalizationModel()
+
+# Inicializace
+
+
+
+# Textové vstupní pole
+input_text = st.text_input("Zadejte text zde:")
+
+# Tlačítko pro potvrzení
+submit_button = st.button("Odeslat")
+
+# Akce po stisknutí tlačítka
+if submit_button:
+    st.write(f"Výstup: {model.run(input_text)}")

requirements.txt

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