Marmoot's picture
Update app.py
5520c7a
import gradio as gr
from transformers import pipeline
pipe = pipeline("text-classification", model="hackathon-pln-es/electricidad-base-generator-fake-news")
def predict(text):
return pipe(text)[0]["text-classification_text"]
title = "Detección de FakeNews en Español"
iface = gr.Interface(
fn=predict,
inputs=[gr.inputs.Textbox(label="text", lines=3)],
outputs='text',
title=title,
examples=[["La fiesta en Sevilla por el vuelco electoral se alargó hasta casi las dos de la madrugada. La algarabía se desbordó en las calles de la capital andaluza. La marcha militar ""Soldadito español"" de Jacinto Guerrero Torres puso la nota musical a la velada."]]
)
iface.launch(debug=True)