Update app.py

app.py

@@ -1,21 +1,4 @@
 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()
+gr.load("models/Devarshi/Brain_Tumor_Classification").launch()