KvrParaskevi commited on
Commit
6ba3e77
·
verified ·
1 Parent(s): b58ca25

Rename chatbot_bedrock.py to chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot_bedrock.py → chatbot.py +2 -0
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