peterpeter8585 commited on
Commit
c78e2fb
·
verified ·
1 Parent(s): 2c98113

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -12
app.py CHANGED
@@ -78,8 +78,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
78
  #m=M.from_pretrained("peterpeter8585/syai4.3")
79
  #t=T.from_pretrained("peterpeter8585/syai4.3")
80
  #pipe=pipeline(model=m,tokenizer=t,task="text-generation")
81
- llm=HuggingFacePipeline.from_model_id(model_id="peterpeter8585/syaideepcoder1",task="text-generation")
82
- llm=ChatHuggingFace(llm=llm)
83
  from langchain.retrievers import WikipediaRetriever as Wiki
84
  import gradio as gr
85
  chatbot = gr.Chatbot(
@@ -96,23 +95,16 @@ tools.append(YTS())
96
  tools.extend(load_tools(["requests_all"],allow_dangerous_tools=True))
97
  tools.extend(load_tools(["llm-math","ddg-search"],llm=llm))
98
  tools.append(Tool.from_function(func=terminal,name="terminal",description="터미널 명령어실행에 적합함"))
 
99
  tools.append(crt(name="wiki",description="위키 백과를 검색하여 정보를 가져온다",retriever=Wiki(lang="ko",top_k_results=1)))
100
- agent=Ex(agent=Agent(llm,tools,prompt),tools=tools,verbose=True,handle_parsing_errors=True)
101
  def chat(message,
102
  history: list[tuple[str, str]],
103
  system_message,
104
  max_tokens,
105
  temperature,
106
  top_p, chat_session):
107
- messages=[]
108
- for val in history:
109
- if val[0]:
110
- messages.append(val[0])
111
- if val[1]:
112
- messages.append(val[1])
113
-
114
- messages.append(message)
115
- return agent.invoke({"input":messages})
116
  ai1=gr.ChatInterface(
117
  chat,
118
  chatbot=chatbot,
 
78
  #m=M.from_pretrained("peterpeter8585/syai4.3")
79
  #t=T.from_pretrained("peterpeter8585/syai4.3")
80
  #pipe=pipeline(model=m,tokenizer=t,task="text-generation")
81
+ llm=HuggingFacePipeline.from_model_id(model_id="peterpeter8585/llama3_1",task="text-generation")
 
82
  from langchain.retrievers import WikipediaRetriever as Wiki
83
  import gradio as gr
84
  chatbot = gr.Chatbot(
 
95
  tools.extend(load_tools(["requests_all"],allow_dangerous_tools=True))
96
  tools.extend(load_tools(["llm-math","ddg-search"],llm=llm))
97
  tools.append(Tool.from_function(func=terminal,name="terminal",description="터미널 명령어실행에 적합함"))
98
+ memory=MEM()
99
  tools.append(crt(name="wiki",description="위키 백과를 검색하여 정보를 가져온다",retriever=Wiki(lang="ko",top_k_results=1)))
100
+ agent=Ex(agent=Agent(llm,tools,prompt),tools=tools,verbose=True,handle_parsing_errors=True,memory=memory)
101
  def chat(message,
102
  history: list[tuple[str, str]],
103
  system_message,
104
  max_tokens,
105
  temperature,
106
  top_p, chat_session):
107
+ return agent.invoke({"input":message})
 
 
 
 
 
 
 
 
108
  ai1=gr.ChatInterface(
109
  chat,
110
  chatbot=chatbot,