update stuff

app.py

@@ -32,6 +32,7 @@ model.load_state_dict(state_dict, strict=False)
 model.eval()
 
 def predict(im):
+    im = im['background']
     print(im)
     x = torch.tensor(im, dtype=torch.float32).unsqueeze(0).unsqueeze(0) / 255.