Update app.py

app.py

@@ -17,7 +17,10 @@ import gradio as gr
 
 # Cargamos el learner
 #learn = load_learner('export.pkl')
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu") 
 model = torch.jit.load("unet.pth")
+model = model.cpu()
+model.eval()
 
 # Definimos las etiquetas de nuestro modelo
 #labels = learn.dls.vocab