Rename chatbot_bedrock.py to chatbot.py
Browse files
chatbot_bedrock.py → chatbot.py
RENAMED
@@ -8,6 +8,7 @@ import streamlit as st
|
|
8 |
from langchain_community.llms import HuggingFaceHub
|
9 |
|
10 |
my_model_id = os.getenv('MODEL_REPO_ID', 'Default Value')
|
|
|
11 |
|
12 |
@st.cache_resource
|
13 |
def load_model():
|
@@ -22,6 +23,7 @@ def load_model():
|
|
22 |
"temperature": 0.1,
|
23 |
"repetition_penalty": 1.03,
|
24 |
},
|
|
|
25 |
)
|
26 |
return model
|
27 |
|
|
|
8 |
from langchain_community.llms import HuggingFaceHub
|
9 |
|
10 |
my_model_id = os.getenv('MODEL_REPO_ID', 'Default Value')
|
11 |
+
token = os.getenv('HUGGINGFACEHUB_API_TOKEN')
|
12 |
|
13 |
@st.cache_resource
|
14 |
def load_model():
|
|
|
23 |
"temperature": 0.1,
|
24 |
"repetition_penalty": 1.03,
|
25 |
},
|
26 |
+
token = token,
|
27 |
)
|
28 |
return model
|
29 |
|