KalooAI commited on
Commit
eafbe54
·
verified ·
1 Parent(s): 9b8de6a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -62,7 +62,8 @@ from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
62
 
63
  # 1. Create the endpoint with the 'conversational' task
64
  llm_endpoint = HuggingFaceEndpoint(
65
- repo_id="mistralai/Mistral-7B-Instruct-v0.3",
 
66
  task="conversational", # Change this from 'text-generation'
67
  max_new_tokens=512,
68
  huggingfacehub_api_token=os.getenv("HF_TOKEN"),
 
62
 
63
  # 1. Create the endpoint with the 'conversational' task
64
  llm_endpoint = HuggingFaceEndpoint(
65
+ #repo_id="mistralai/Mistral-7B-Instruct-v0.3",
66
+ repo_id = "HuggingFaceH4/zephyr-7b-beta",
67
  task="conversational", # Change this from 'text-generation'
68
  max_new_tokens=512,
69
  huggingfacehub_api_token=os.getenv("HF_TOKEN"),