peterpeter8585 commited on
Commit
d940786
·
verified ·
1 Parent(s): 723e098

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from langchain_huggingface import HuggingFaceEndpoint as HF
2
  from subprocess import Popen, PIPE as P
3
  from langchain_experimental.tools.python.tool import PythonREPLTool as PYT
4
  from langchain.agents import load_tools, create_structured_chat_agent as Agent,AgentExecutor as Ex, AgentType as Type
@@ -62,6 +62,9 @@ prompt = ChatPromptTemplate.from_messages(
62
  ("human", human),
63
  ]
64
  )
 
 
 
65
  from langchain.retrievers import WikipediaRetriever as Wiki
66
  import gradio as gr
67
  chatbot = gr.Chatbot(
@@ -72,7 +75,6 @@ chatbot = gr.Chatbot(
72
  def terminal(c):
73
  a=Popen(c,shell=True,stdin=P,stdout=P,stderr=P)
74
  return a.stdout.read()+a.stderr.read()
75
- llm=HF(repo_id="peterpeter8585/syai4.1")
76
  tools=FMT().get_tools()
77
  tools.append(PYT())
78
  tools.extend(load_tools(["requests_all"],allow_dangerous_tools=True))
 
1
+ from langchain_huggingface import HuggingFacePipeline as HF
2
  from subprocess import Popen, PIPE as P
3
  from langchain_experimental.tools.python.tool import PythonREPLTool as PYT
4
  from langchain.agents import load_tools, create_structured_chat_agent as Agent,AgentExecutor as Ex, AgentType as Type
 
62
  ("human", human),
63
  ]
64
  )
65
+ from transformers import pipeline
66
+ pipe=pipeline(task="textgeneration",model="peterpeter8585/syai4.1")
67
+ llm=HF(pipeline=pipe)
68
  from langchain.retrievers import WikipediaRetriever as Wiki
69
  import gradio as gr
70
  chatbot = gr.Chatbot(
 
75
  def terminal(c):
76
  a=Popen(c,shell=True,stdin=P,stdout=P,stderr=P)
77
  return a.stdout.read()+a.stderr.read()
 
78
  tools=FMT().get_tools()
79
  tools.append(PYT())
80
  tools.extend(load_tools(["requests_all"],allow_dangerous_tools=True))