people_mate / app.py
zeyadahmedd's picture
Update app.py
afc63ec
raw
history blame
1.12 kB
import gradio as gr
from langchain import LlamaCpp
from langchain.callbacks import StreamingStdOutCallbackHandler
from langchain.callbacks.manager import CallbackManager
from langchain.chains.question_answering import load_qa_chain
import dill
def greet(name):
db = dill.load(open("client_embebedings.mate", "rb"))
docs = db.similarity_search(name)
return chain.run(input_documents=docs, question=name)
def greet1(name):
return "Hello ya client" +name
iface = gr.Blocks()
with iface:
callback_manager = CallbackManager([StreamingStdOutCallbackHandler()])
chain = load_qa_chain( LlamaCpp(
model_path="llama-13B-Q4_K_M.gguf",
temperature=0,
top_p= 1,
callback_manager=callback_manager,
verbose=True,
), chain_type="stuff")
name = gr.Textbox(label="Name")
output = gr.Textbox(label="Output Box")
greet_btn = gr.Button("Greet")
greet_btn.click(fn=greet, inputs=name, outputs=output, api_name="greet")
greet1_btn = gr.Button("Greet1")
greet1_btn.click(fn=greet1, inputs=name, outputs=output, api_name="testing")
iface.launch()