ssaiteja16's picture
Update app.py
4db3795 verified
raw
history blame
393 Bytes
import gradio as gr
from model import generate_response
from datasets import load_dataset
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()