peterpeter8585 commited on
Commit
a5d1f7e
·
verified ·
1 Parent(s): d846f01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from langchain_huggingface import HuggingFacePipeline as HF,ChatHuggingFace
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,10 +62,11 @@ prompt = ChatPromptTemplate.from_messages(
62
  ("human", human),
63
  ]
64
  )
65
- from transformers import pipeline
66
- pipe=pipeline(model="peterpeter8585/syai4.3",task="text-generation")
 
 
67
  llm=HF(pipeline=pipe)
68
- llm=ChatHuggingFace(llm=llm)
69
  from langchain.retrievers import WikipediaRetriever as Wiki
70
  import gradio as gr
71
  chatbot = gr.Chatbot(
 
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,AutoModelForCausalLM as M,AutoTokenizer as T
66
+ m=M.from_pretrained("peterpeter8585/syai4.3")
67
+ t=T.from_pretrained("peterpeter8585/syai4.3")
68
+ pipe=pipeline(model=m,tokenizer=t,task="text-generation")
69
  llm=HF(pipeline=pipe)
 
70
  from langchain.retrievers import WikipediaRetriever as Wiki
71
  import gradio as gr
72
  chatbot = gr.Chatbot(