peerawatchomp commited on
Commit
d768f9c
·
verified ·
1 Parent(s): 9ad864a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -8
app.py CHANGED
@@ -27,17 +27,14 @@ class BasicAgent:
27
  def __init__(self):
28
  print("BasicAgent initialized.")
29
 
 
 
30
  # llm = HuggingFaceEndpoint(
31
- # repo_id="mistralai/Mistral-7B-v0.1",
32
- # huggingfacehub_api_token=os.environ.get("HUGGINGFACEHUB_API_TOKEN"),
33
  # )
34
-
35
- llm = HuggingFaceEndpoint(
36
- repo_id="mistralai/Mistral-7B-Instruct-v0.2",
37
- huggingfacehub_api_token=os.environ.get("HUGGINGFACEHUB_API_TOKEN"),
38
- )
39
 
40
- chat = ChatHuggingFace(llm=llm, verbose=True)
41
 
42
  # Set up tools
43
  search_tool = DuckDuckGoSearchRun()
 
27
  def __init__(self):
28
  print("BasicAgent initialized.")
29
 
30
+ chat = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
31
+
32
  # llm = HuggingFaceEndpoint(
33
+ # repo_id="mistralai/Mistral-7B-Instruct-v0.2",
34
+ # huggingfacehub_api_token=os.environ.get("HUGGINGFACEHUB_API_TOKEN"),
35
  # )
 
 
 
 
 
36
 
37
+ # chat = ChatHuggingFace(llm=llm, verbose=True)
38
 
39
  # Set up tools
40
  search_tool = DuckDuckGoSearchRun()