Harika22 commited on
Commit
792a566
·
verified ·
1 Parent(s): 987110a

Update pages/model.py

Browse files
Files changed (1) hide show
  1. pages/model.py +4 -3
pages/model.py CHANGED
@@ -10,7 +10,8 @@ from langchain.chains import LLMChain
10
  from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
11
 
12
 
13
- os.environ["HUGGINGFACEHUB_API_KEY"] = os.getenv("HF_TOKEN")
 
14
 
15
  st.set_page_config(page_title="🧠 MediAssist", layout="centered")
16
 
@@ -56,8 +57,8 @@ if uploaded_file:
56
 
57
  llm = ChatHuggingFace(
58
  llm=HuggingFaceEndpoint(
59
- repo_id="mistralai/Mistral-7B-Instruct-v0.3",
60
- provider="novita",
61
  temperature=0.5,
62
  max_new_tokens=500,
63
  task="conversational"
 
10
  from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
11
 
12
 
13
+ os.environ["HF_TOKEN"]=os.getenv('HF_Token')
14
+ os.environ["HUGGINGFACEHUB_API_KEY"]=os.getenv('HF_Token')
15
 
16
  st.set_page_config(page_title="🧠 MediAssist", layout="centered")
17
 
 
57
 
58
  llm = ChatHuggingFace(
59
  llm=HuggingFaceEndpoint(
60
+ repo_id="meta-llama/Llama-3.1-8B-Instruct",
61
+ provider="nebius",
62
  temperature=0.5,
63
  max_new_tokens=500,
64
  task="conversational"