File size: 559 Bytes
c3125e8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
from dotenv import load_dotenv
from langchain.embeddings import HuggingFaceHub, LLMChain
from langchain.prompts import prompt_templates
import gradio
load_dotenv()
os.getenv('HF_API')
hub_llm = HuggingFaceHub(repo_id='facebook/blenderbot-400M-distill')
prompt = prompt_templates(
input_variable = ["question"]
)
hub_chain = LLMChain(prompt=prompt, llm=hub_llm, verbose=True)
def responsenew(data):
return hub_chain(data)
gradio_interface = gradio.Interface(
fn = responsenew,
inputs = "text",
outputs = "text"
)
gradio_interface.launch()
|