awacke1 commited on
Commit
b74139a
·
1 Parent(s): c0a6fa6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -153,7 +153,8 @@ def readitaloud(result):
153
  def chat_with_model(prompt, document_section, model_choice='Llama-2-7b-chat-hf'):
154
  # API_URL = 'https://qe55p8afio98s0u3.us-east-1.aws.endpoints.huggingface.cloud' # Dr Llama
155
  API_URL = "https://api-inference.huggingface.co/models/meta-llama/Llama-2-7b-chat-hf" # HF default model for l7b
156
- API_KEY = os.getenv('API_KEY')
 
157
  MODEL1="meta-llama/Llama-2-7b-chat-hf"
158
  MODEL1URL="https://huggingface.co/meta-llama/Llama-2-7b-chat-hf"
159
  HF_KEY = os.getenv('HF_KEY')
 
153
  def chat_with_model(prompt, document_section, model_choice='Llama-2-7b-chat-hf'):
154
  # API_URL = 'https://qe55p8afio98s0u3.us-east-1.aws.endpoints.huggingface.cloud' # Dr Llama
155
  API_URL = "https://api-inference.huggingface.co/models/meta-llama/Llama-2-7b-chat-hf" # HF default model for l7b
156
+ #API_KEY = os.getenv('API_KEY')
157
+ API_KEY = os.getenv('HF_KEY')
158
  MODEL1="meta-llama/Llama-2-7b-chat-hf"
159
  MODEL1URL="https://huggingface.co/meta-llama/Llama-2-7b-chat-hf"
160
  HF_KEY = os.getenv('HF_KEY')