Update app.py

app.py

@@ -3,8 +3,10 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 from datasets import load_dataset
 import spaces
+from huggingface-hub import login
+
+login(token=hf_token)
 
-# Load model once at startup
 try:
     model = AutoModelForCausalLM.from_pretrained("flamiry/first")
     tokenizer = AutoTokenizer.from_pretrained("flamiry/first")
@@ -16,11 +18,9 @@ tokenizer.pad_token = tokenizer.eos_token
 @spaces.GPU
 def train_model():
     try:
-        # Load Slovak data
         dataset = load_dataset("DGurgurov/slovak_sa", split="train")
         slovak_texts = list(dataset['text'])[:50]
         
-        # Tokenize
         inputs = tokenizer(
             slovak_texts,
             return_tensors="pt",
@@ -29,7 +29,6 @@ def train_model():
             max_length=64
         )
         
-        # Train
         optimizer = torch.optim.Adam(model.parameters(), lr=5e-5)
         for epoch in range(2):
             optimizer.zero_grad()
@@ -54,7 +53,6 @@ def generate_text(prompt):
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
-# Create interface
 with gr.Blocks() as demo:
     gr.Markdown("# 🇸🇰 Slovak LLM Training")