Update app.py

app.py

@@ -1,8 +1,21 @@
 import gradio as gr
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
 
-# Cargar el modelo
-modelo = gr.load("models/Devarshi/Brain_Tumor_Classification")
+# Cargar el modelo desde Hugging Face
+modelo = AutoModelForSequenceClassification.from_pretrained("Devarshi/Brain_Tumor_Classification")
+tokenizer = AutoTokenizer.from_pretrained("Devarshi/Brain_Tumor_Classification")
 
-# Crear la interfaz gráfica sin mostrar el título
-interfaz = gr.Interface(modelo, live=True, show_live=True, show_interpreter=True)
+# 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()
+