Update app.py

app.py

@@ -1,14 +1,22 @@
 import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from huggingface_hub import login
+import os
+
+# Authenticate with HuggingFace using environment variable
+hf_token = os.getenv('HF_TOKEN')
+if hf_token:
+    login(token=hf_token)
 
 # Load FunctionGemma model and tokenizer
 model_name = "google/functiongemma-270m-it"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token)
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
     torch_dtype=torch.float16,
-    device_map="auto"
+    device_map="auto",
+    token=hf_token
 )
 
 def test_function_calling(user_input: str, temperature: float = 0.7) -> str: