import gradio as gr
from transformers import T5Tokenizer, T5ForConditionalGeneration

tokenizer = T5Tokenizer.from_pretrained("cuneytkaya/fintech-chatbot-t5")
model = T5ForConditionalGeneration.from_pretrained("cuneytkaya/fintech-chatbot-t5")

def chatbot(input_text):
    inputs = tokenizer(input_text, return_tensors="pt")
    outputs = model.generate(inputs.input_ids)
    return tokenizer.decode(outputs[0], skip_special_tokens=True)

interface = gr.Interface(fn=chatbot, 
                         inputs="text", 
                         outputs="text",
                         title="Fintech Chatbot",
                         description="This chatbot answers common fintech and banking-related queries.")

interface.launch(share=True)