Update main.py
Browse files
main.py
CHANGED
@@ -922,7 +922,7 @@ async def API_connexion(romeListArray):
|
|
922 |
@cl.step(type="llm")
|
923 |
async def IA():
|
924 |
os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
|
925 |
-
repo_id = "mistralai/Mixtral-
|
926 |
|
927 |
llm = HuggingFaceEndpoint(
|
928 |
repo_id=repo_id, max_new_tokens=8000, temperature=0.5, streaming=True
|
@@ -983,7 +983,7 @@ async def setup_agent(settings):
|
|
983 |
contextChat = "Il n'y a pas de contexte."
|
984 |
|
985 |
os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
|
986 |
-
repo_id = "mistralai/Mixtral-
|
987 |
|
988 |
model = HuggingFaceEndpoint(
|
989 |
repo_id=repo_id, max_new_tokens=8000, temperature=0.5, streaming=True
|
|
|
922 |
@cl.step(type="llm")
|
923 |
async def IA():
|
924 |
os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
|
925 |
+
repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
926 |
|
927 |
llm = HuggingFaceEndpoint(
|
928 |
repo_id=repo_id, max_new_tokens=8000, temperature=0.5, streaming=True
|
|
|
983 |
contextChat = "Il n'y a pas de contexte."
|
984 |
|
985 |
os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
|
986 |
+
repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
987 |
|
988 |
model = HuggingFaceEndpoint(
|
989 |
repo_id=repo_id, max_new_tokens=8000, temperature=0.5, streaming=True
|