Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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/
|
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 |
-
|
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,
|