Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
|
@@ -49,7 +49,8 @@ def auth_callback(username: str, password: str):
|
|
| 49 |
@cl.step(type="tool")
|
| 50 |
async def LLModel():
|
| 51 |
os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
|
| 52 |
-
repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
|
|
|
| 53 |
llm = HuggingFaceEndpoint(
|
| 54 |
repo_id=repo_id, max_new_tokens=5300, temperature=0.5, task="text2text-generation", streaming=True
|
| 55 |
)
|
|
|
|
| 49 |
@cl.step(type="tool")
|
| 50 |
async def LLModel():
|
| 51 |
os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
|
| 52 |
+
#repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
| 53 |
+
repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
|
| 54 |
llm = HuggingFaceEndpoint(
|
| 55 |
repo_id=repo_id, max_new_tokens=5300, temperature=0.5, task="text2text-generation", streaming=True
|
| 56 |
)
|