Sannidhi commited on
Commit
898c540
·
verified ·
1 Parent(s): dec8af2

Update predict_stress.py

Browse files
Files changed (1) hide show
  1. predict_stress.py +1 -1
predict_stress.py CHANGED
@@ -13,7 +13,7 @@ xgboostmodel_id = "Sannidhi/stress_prediction_xgboost_model"
13
  xgboost_model = None
14
  generator = pipeline("text-generation", model=model_id)
15
 
16
- def get_llm_response(prompt_text, model_id="Sasmitah/llama_16bit_2", max_new_tokens=256, temperature=0.5):
17
  """Generates a response from the Hugging Face model for a given prompt text."""
18
  try:
19
  llm = HuggingFaceEndpoint(
 
13
  xgboost_model = None
14
  generator = pipeline("text-generation", model=model_id)
15
 
16
+ def get_llm_response(prompt_text, model_id="meta-llama/Llama-3.2-1B-Instruct", max_new_tokens=256, temperature=0.5):
17
  """Generates a response from the Hugging Face model for a given prompt text."""
18
  try:
19
  llm = HuggingFaceEndpoint(