CCockrum commited on
Commit
57f589e
·
verified ·
1 Parent(s): 546ff54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ if "follow_up" not in st.session_state:
33
  model_id = "mistralai/Mistral-7B-Instruct-v0.3"
34
 
35
  # ✅ Initialize Hugging Face Model
36
- def get_llm_hf_inference(model_id=model_id, max_new_tokens=128, temperature=0.7):
37
  return HuggingFaceEndpoint(
38
  repo_id=model_id,
39
  max_new_tokens=max_new_tokens,
 
33
  model_id = "mistralai/Mistral-7B-Instruct-v0.3"
34
 
35
  # ✅ Initialize Hugging Face Model
36
+ def get_llm_hf_inference(model_id=model_id, max_new_tokens=1024, temperature=0.7):
37
  return HuggingFaceEndpoint(
38
  repo_id=model_id,
39
  max_new_tokens=max_new_tokens,