Update health_tracker_agent.py

health_tracker_agent.py

@@ -1,11 +1,17 @@
 from transformers import AutoTokenizer, AutoModelForCausalLM
-import torch
+from huggingface_hub import login
+
+def track_health_status(input_text: str) -> str:
+    # Authenticate
+    login("your_huggingface_token_here")  # <-- replace with your actual token
 
-def track_health_status(health_input):
     model_name = "m42-health/med42-70b"
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name)
-    prompt = f"Based on the following health input, track possible issues or advice: {health_input}."
-    inputs = tokenizer(prompt, return_tensors="pt")
-    outputs = model.generate(**inputs, max_new_tokens=150)
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+    tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
+    model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=True)
+
+    inputs = tokenizer.encode(input_text, return_tensors="pt", truncation=True)
+    outputs = model.generate(inputs, max_length=512)
+    result = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    
+    return result