pradeep4321 commited on
Commit
6899cb0
·
verified ·
1 Parent(s): d5ca8d5

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -20,7 +20,7 @@ st.title("🏢 Company AI Assistant")
20
  def load_models():
21
  embed_model = SentenceTransformer("all-MiniLM-L6-v2")
22
  llm = InferenceClient(
23
- model="meta-llama/Llama-3-8b-instruct",
24
  token=os.environ.get("HF_TOKEN")
25
  )
26
  return embed_model, llm
 
20
  def load_models():
21
  embed_model = SentenceTransformer("all-MiniLM-L6-v2")
22
  llm = InferenceClient(
23
+ model="meta-llama/Meta-Llama-3-8B-Instruct",
24
  token=os.environ.get("HF_TOKEN")
25
  )
26
  return embed_model, llm