Update health_tracker_agent.py

health_tracker_agent.py

@@ -1,19 +1,16 @@
-import os
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from huggingface_hub import login
+import os
 
 def track_health_status(input_text: str) -> str:
-    # Get token from environment variable
+    # Secure way to load token
     hf_token = os.getenv("HF_TOKEN")
-    if not hf_token:
-        raise ValueError("Hugging Face token not set in environment variable 'HF_TOKEN'")
-
     login(hf_token)
 
     model_name = "m42-health/med42-70b"
 
-    tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
-    model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=True)
+    tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token)
+    model = AutoModelForCausalLM.from_pretrained(model_name, token=hf_token)
 
     inputs = tokenizer.encode(input_text, return_tensors="pt", truncation=True)
     outputs = model.generate(inputs, max_length=512)