update

app.py

@@ -7,6 +7,8 @@ import plotly.graph_objects as go
 from PIL import Image
 import spaces
 import matplotlib.cm as cm
+import time
+
 
 @spaces.GPU
 def process_images(image1, image2):
@@ -25,8 +27,9 @@ def process_images(image1, image2):
         "Model is on device: ",
         model.device,
         "and inputs are on device: ",
-        inputs.device,
+        inputs["pixel_values"].device,
     )
+
     with torch.no_grad():
         outputs = model(**inputs)
 
@@ -165,7 +168,7 @@ with gr.Blocks(title="LightGlue Matching Demo") as demo:
     output_plot = gr.Plot(label="Matching Results", scale=2)
 
     # Connect the function
-    process_btn.click(fn=process_images, inputs=[image1, image2], outputs=output_plot)
+    process_btn.click(fn=process_images, inputs=[image1, image2], outputs=[output_plot])
 
     # Add some example usage