Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from langchain import PromptTemplate, LLMChain | |
| from langchain_huggingface import HuggingFacePipeline, HuggingFaceEndpoint | |
| from transformers import pipeline | |
| import os | |
| os.environ["HUGGINGFACEHUB_API_TOKEN"] | |
| pipe = pipeline( | |
| 'text2text-generation', | |
| model='VMware/flan-t5-large-alpaca', | |
| max_length=60, | |
| do_sample=True, | |
| temperature=0.9 | |
| ) | |
| llm = HuggingFacePipeline(pipeline=pipe) | |
| prompt_template = """ you are a highly knownlegdable and helpful AI assistant.Engage in a conversation with the user.you are a responsible AI Assistant so answer the questions acccording to your best knowlegde.your main is to provide clear and informative answer to the user questions. | |
| User: {question} | |
| Assistant:""" | |
| prompt = PromptTemplate(template=prompt_template, input_variables=["question"]) | |
| chain = LLMChain(llm=llm, prompt=prompt) | |
| def chatbot(question,chat_history): | |
| response = chain.run(question) | |
| return response | |
| demo = gr.ChatInterface( | |
| fn=chatbot, | |
| title="Chatbot", | |
| description="Helpful AI Assistant!!" | |
| ) | |
| demo.launch() |