Update app.py

app.py

@@ -18,7 +18,7 @@ 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("modelResnet18.pth")
+model = torch.jit.load("unet.pth")
 model = model.cpu()
 model.eval()