Spaces:
Sleeping
Sleeping
Update pages/model.py
Browse files- 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["
|
|
|
|
| 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="
|
| 60 |
-
provider="
|
| 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"
|