RESTRA / app.py
Adriiiii24's picture
Update app.py
8bc5b1d verified
raw
history blame
691 Bytes
import gradio as gr
from transformers import pipeline, T5Tokenizer, T5ForConditionalGeneration
import os
token = os.getenv("HF_TOKEN")
tokenizer = T5Tokenizer.from_pretrained("sumedh/t5-base-amazonreviews", clean_up_tokenization_spaces=True)
model = T5ForConditionalGeneration.from_pretrained("sumedh/t5-base-amazonreviews")
summarizer = pipeline("summarization", model="sumedh/t5-base-amazonreviews")
def texto_sum(text):
summary = summarizer(text, do_sample=False)
return summary[0]['summary_text']
demo = gr.Interface(
fn=texto_sum,
inputs=gr.Textbox(label="Texto a introducir:", placeholder="Introduce el texto a resumir aquí..."),
outputs="text"
)
demo.launch()