Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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,
|