Update app.py

app.py

@@ -1,16 +1,19 @@
+import os
+
 from typing import Literal, Optional, Tuple
-from pathlib import Path
 import logging
 
 import gradio as gr
 from omegaconf import OmegaConf
 from dacite import Config as DaciteConfig, from_dict
 from transformers import GPT2Config, GPT2LMHeadModel
-from huggingface_hub import PyTorchModelHubMixin
+from huggingface_hub import PyTorchModelHubMixin, login
 
 from llm_trainer import LLMTrainer
 from xlstm import xLSTMLMModel, xLSTMLMModelConfig
 
+login(token=os.getenv('token'))
+
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)