File size: 691 Bytes
8a4e829 45ba841 8a4e829 45ba841 6b797a9 45ba841 6b797a9 45ba841 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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()
|