Update app.py

app.py

@@ -1,13 +1,11 @@
 from fastai.vision.all import *
 import gradio as gr
 import torchvision.transforms as transforms
+import torch
 
 # Cargamos el learner
 learn = load_learner('best_model.pkl')
 
-# Definimos las etiquetas de nuestro modelo
-labels = learn.dls.vocab
-
 def transform_image(image, device):
     my_transforms = transforms.Compose([transforms.ToTensor(),
                                         transforms.Normalize(