| from dotenv import load_dotenv |
| from langchain import HuggingFaceHub, LLMChain |
| from langchain.prompts import PromptTemplates |
| import gradio |
|
|
| load_dotenv() |
| os.getenv('HF_API') |
|
|
| hub_llm = HuggingFaceHub(repo_id='facebook/blenderbot-400M-distill') |
|
|
| prompt = prompt_templates( |
| input_variable = ["question"], |
| template = "Answer is: {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() |
|
|
