Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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"),
|