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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -150,8 +150,7 @@ def readitaloud(result):
150
  components.html(documentHTML5, width=800, height=300)
151
  #return result
152
 
153
- # Llama 7b
154
- def chat_with_model(prompt, document_section, model_choice='gpt-3.5-turbo'):):
155
  # API_URL = 'https://qe55p8afio98s0u3.us-east-1.aws.endpoints.huggingface.cloud' # Dr Llama
156
  API_URL = "https://api-inference.huggingface.co/models/meta-llama/Llama-2-7b-chat-hf" # HF default model for l7b
157
  API_KEY = os.getenv('API_KEY')
 
150
  components.html(documentHTML5, width=800, height=300)
151
  #return result
152
 
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')