ssaiteja16's picture
Update app.py
a21e182 verified
raw
history blame
407 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="Capstone Project Group 10")
if __name__ == "__main__":
iface.launch()