Update app.py

app.py

@@ -4,6 +4,9 @@ from PIL import Image
 from transformers import (VisionEncoderDecoderModel, AutoTokenizer, AutoImageProcessor,
                           BitsAndBytesConfig)
 
+from huggingface_hub import login
+login(token=os.environ["HF_TOKEN"])
+
 MODEL_ID = os.getenv("MODEL_ID", "Parsa2025AI/r2gen-swin-cerebras")
 GEN_MAX_LEN = int(os.getenv("GEN_MAX_LEN", "192"))
 NUM_BEAMS = int(os.getenv("NUM_BEAMS", "1"))