Spaces:
Runtime error
Runtime error
import gradio as gr | |
from langchain_core.prompts import ChatPromptTemplate | |
from langchain.chains.combine_documents import create_stuff_documents_chain | |
from langchain.chains import create_retrieval_chain | |
from langchain import HuggingFaceHub | |
import os | |
os.environ["HUGGINGFACEHUB_API_TOKEN"] = os.getenv('hf_token') | |
llm = HuggingFaceHub(repo_id="tiiuae/falcon-7b-instruct", | |
model_kwargs={"max_length": 300, "max_new_tokens": 300}) | |
def echo(message, history): | |
message = llm.invoke(message) | |
return message | |
demo = gr.ChatInterface(fn=echo, examples=["hello", "hola", "merhaba"], title="Echo Bot") | |
demo.launch() |