File size: 393 Bytes
71e2c65
 
59284b1
 
4c66f77
 
 
 
 
71e2c65
 
4c66f77
71e2c65
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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()