Spaces:

JamalAG
/

Agent

Sleeping

JamalAG commited on Jun 23

Commit

9fc789d

•

1 Parent(s): 68de2dc

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -8,7 +8,7 @@ def query_agent(data, query):
     # Parse the CSV file and create a Pandas DataFrame from its contents.
     df = pd.read_csv(data)
-    llm = HuggingFaceEndpoint(repo_id="teknium/OpenHermes-2.5-Mistral-7B",
                               temperature = 0.9,
                               stop_sequences =  ["\nQuestion:"])

     # Parse the CSV file and create a Pandas DataFrame from its contents.
     df = pd.read_csv(data)
+    llm = HuggingFaceEndpoint(repo_id="HuggingFaceH4/zephyr-7b-beta",
                               temperature = 0.9,
                               stop_sequences =  ["\nQuestion:"])