Update predict_stress.py

predict_stress.py

@@ -11,10 +11,11 @@ from langchain_core.output_parsers import StrOutputParser
 
 xgboostmodel_id = "Sannidhi/stress_prediction_xgboost_model"
 xgboost_model = None
+model_id="meta-llama/Llama-3.2-1B-Instruct"
 generator = pipeline("text-generation", model=model_id)
-model_id="unsloth/Llama-3.2-1B-Instruct"
 
-def get_llm_response(prompt_text, model_id="unsloth/Llama-3.2-1B-Instruct", max_new_tokens=256, temperature=0.5):
+
+def get_llm_response(prompt_text, model_id="meta-llama/Llama-3.2-1B-Instruct", max_new_tokens=256, temperature=0.5):
     """Generates a response from the Hugging Face model for a given prompt text."""
     try:
         llm = HuggingFaceEndpoint(