Spaces:
Runtime error
Runtime error
File size: 916 Bytes
103a097 1065e57 c2ba8e6 103a097 1065e57 c2ba8e6 1065e57 103a097 1065e57 103a097 1065e57 cdb3da1 c2ba8e6 cdb3da1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import gradio as gr
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
import torch
# Load the BART tokenizer and model
tokenizer = AutoTokenizer.from_pretrained("facebook/bart-base")
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-base")
def generate_response(user_input):
inputs = tokenizer(user_input, return_tensors="pt", max_length=512, truncation=True)
outputs = model.generate(**inputs)
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
return response
input_textbox = gr.Textbox(lines=10, label="Enter your text here")
output_textbox = gr.outputs.Textbox(label="Chatbot Response")
chatbot_interface = gr.Interface(fn=generate_response, inputs=input_textbox, outputs=output_textbox, title="Hugging Face BART Chatbot", description="This chatbot uses the Hugging Face BART model to generate responses based on user input.")
chatbot_interface.launch()
|