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()