Update main.py

main.py

@@ -4,12 +4,14 @@ from typing import Optional
 from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 from fastapi.middleware.cors import CORSMiddleware
 from datasets import load_dataset
-
+import os
 
 # ==== Load tokenizer & model Vit5 fine-tune ====
-model_name = "lingling707/vit5-skinbot"  # checkpoint fine-tune
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+model_name = "lingling707/vit5-skinbot"
+hf_token = os.getenv("HF_TOKEN")
+
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name, use_auth_token=hf_token)
 
 # ==== Tạo pipeline chatbot ====
 chatbot = pipeline(