Spaces:
Sleeping
Sleeping
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() | |