Update src/streamlit_app.py
Browse files- src/streamlit_app.py +2 -2
src/streamlit_app.py
CHANGED
@@ -5,6 +5,7 @@ import os
|
|
5 |
# constants
|
6 |
QUESTION = "Compute the integral of f(x) = x^2."
|
7 |
MODEL = "Qwen/Qwen3-Embedding-0.6B"
|
|
|
8 |
|
9 |
# remembers session
|
10 |
if "response" not in st.session_state:
|
@@ -26,8 +27,7 @@ def get_llm(model_id=MODEL, max_new_tokens=130, temperature=0.7):
|
|
26 |
llm = HuggingFaceEndpoint(
|
27 |
repo_id=model_id,
|
28 |
max_new_tokens=max_new_tokens,
|
29 |
-
temperature=temperature
|
30 |
-
token = os.getenv("HF_TOKEN")
|
31 |
)
|
32 |
return llm
|
33 |
|
|
|
5 |
# constants
|
6 |
QUESTION = "Compute the integral of f(x) = x^2."
|
7 |
MODEL = "Qwen/Qwen3-Embedding-0.6B"
|
8 |
+
hf_token = os.getenv("HF_TOKEN")
|
9 |
|
10 |
# remembers session
|
11 |
if "response" not in st.session_state:
|
|
|
27 |
llm = HuggingFaceEndpoint(
|
28 |
repo_id=model_id,
|
29 |
max_new_tokens=max_new_tokens,
|
30 |
+
temperature=temperature
|
|
|
31 |
)
|
32 |
return llm
|
33 |
|