Update gradio_app.py

gradio_app.py

@@ -77,7 +77,7 @@ def format_output(results):
 @spaces.GPU(duration=10)
 def inference(img, camera_model):
     out = model.calibrate(img.to(device), camera_model=camera_model)
-    save_keys = ["camera", "gravity"] + [f"{k}_uncertainty" for k in ["roll", "pitch", "vfov"]]
+    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()