not-lain's picture
Update app.py
18a2ef6 verified
raw
history blame
805 Bytes
import gradio as gr
def talk(message, history):
return "hi"
DESCRIPTION = """
A rag pipeline with a chatbot feature
Resources used to build this project :
* embedding model : https://huggingface.co/mixedbread-ai/mxbai-embed-large-v1
* dataset : https://huggingface.co/datasets/not-lain/wikipedia-small-3000-embedded (used mxbai-colbert-large-v1 to create the embedding column )
* faiss docs : https://huggingface.co/docs/datasets/v2.18.0/en/package_reference/main_classes#datasets.Dataset.add_faiss_index
* chatbot : https://huggingface.co/google/gemma-7b-it
If you want to support my work consider clicking on the heart react button ❤️🤗
(testing the ui)
"""
demo = gr.ChatInterface(
fn=talk,
chatbot=gr.Chatbot(
),
description=DESCRIPTION,
)
demo.launch(debug=True)