ssaiteja16's picture
Update app.py
4c66f77 verified
raw
history blame
644 Bytes
import gradio as gr
from model import generate_response
from datasets import load_dataset
from langchain.text_splitter import RecursiveCharacterTextSplitter
from langchain.vectorstores import FAISS
from langchain.embeddings import HuggingFaceEmbeddings
from langchain.document_loaders import TextLoader
from langchain.schema import Document
from huggingface_hub import login
from huggingface_hub import whoami
import os
hf_token = os.getenv("HF_TOKEN")
login(hf_token)
def chatbot(prompt):
return whoami
iface = gr.Interface(fn=chatbot, inputs="text", outputs="text", title="My AI Chatbot")
if __name__ == "__main__":
iface.launch()