feministmystique commited on
Commit
4b0e752
·
verified ·
1 Parent(s): ff6a260

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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