Spaces:
Running
Running
Update predict_stress.py
Browse files- 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="
|
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(
|