app.py

import gradio as gr
from transformers import AutoModelForSequenceClassification, AutoTokenizer

model_name = "fertornie/prueba1.0"  # Cambia por tu modelo
model = AutoModelForSequenceClassification.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

def classify_text(text):
    inputs = tokenizer(text, return_tensors="pt")
    outputs = model(**inputs)
    prediction = outputs.logits.argmax(-1).item()
    return "Positivo" if prediction == 1 else "Negativo"

gr.Interface(fn=classify_text, inputs="text", outputs="label").launch()