Update app.py

app.py

@@ -1,10 +1,14 @@
+import os
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
 
-# Load Somali model
-tokenizer = AutoTokenizer.from_pretrained("cmlang/somali-flan-t5")
-model = AutoModelForSeq2SeqLM.from_pretrained("cmlang/somali-flan-t5")
+# Read Hugging Face Token from environment
+hf_token = os.environ.get("HUGGINGFACE_TOKEN")
+
+# Load Somali model (private)
+tokenizer = AutoTokenizer.from_pretrained("cmlang/somali-flan-t5", use_auth_token=hf_token)
+model = AutoModelForSeq2SeqLM.from_pretrained("cmlang/somali-flan-t5", use_auth_token=hf_token)
 
 # Chatbot function
 def chatbot_somali(msg):