Update gradio_app.py

gradio_app.py

@@ -8,6 +8,8 @@ import numpy as np
 import spaces
 import torch
 
+import matplotlib.pyplot as plt
+
 from geocalib import logger, viz2d
 from geocalib.camera import camera_models
 from geocalib.extractor import GeoCalib
@@ -166,7 +168,7 @@ def update_plot(
     fig.canvas.draw()
     img = np.array(fig.canvas.renderer.buffer_rgba())
 
-    fig.close()
+    plt.close()
 
     return img