Update gradio_app.py

gradio_app.py

@@ -80,8 +80,8 @@ def inference(img, camera_model):
     save_keys = ["camera", "gravity"] + [f"{k}_uncertainty" for k in ["roll", "pitch", "vfov", "focal"]]
     res = {k: v.cpu() for k, v in out.items() if k in save_keys}
     # not converting to numpy results in gpu abort
-    res["up_confidence"] = out["up_confidence"].cpu().numpy()
-    res["latitude_confidence"] = out["latitude_confidence"].cpu().numpy()
+    res["up_confidence"] = torch.tensor(out["up_confidence"].cpu().numpy())
+    res["latitude_confidence"] = torch.tensor(out["latitude_confidence"].cpu().numpy())
     return res