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

st.write("""
# Czech Capitalization Model (CCM)
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)}")