import gradio as gr
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

# Cargar el modelo desde Hugging Face
modelo = AutoModelForSequenceClassification.from_pretrained("Devarshi/Brain_Tumor_Classification")
tokenizer = AutoTokenizer.from_pretrained("Devarshi/Brain_Tumor_Classification")

# Definir la función de inferencia
def clasificar_tumor(texto):
    inputs = tokenizer(texto, return_tensors="pt")
    outputs = modelo(**inputs)
    logits = outputs.logits
    return torch.argmax(logits).item()

# Definir la interfaz gráfica
interfaz = gr.Interface(fn=clasificar_tumor, inputs="text", outputs="text")

# Lanzar la interfaz
interfaz.launch()