Update app.py

app.py

@@ -75,7 +75,7 @@ transformer=transforms.Compose([transforms.Resize((480,640)),
 img=transformer(img).unsqueeze(0)
 img=img.cpu()
 
-traced_cell=torch.jit.trace(aux, (img))
+model=torch.jit.trace(aux, (img))
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.cpu()