Update app.py

app.py

@@ -105,10 +105,10 @@ def generate_report(frontal_path, lateral_path, indication, technique, compariso
             get_grounding=grounding
         ).to("cpu")
         
-        # Remove all image size related keys
+        processed = dict(processed)
         image_size_keys = [k for k in processed.keys() if "image_sizes" in k]
         for k in image_size_keys:
-            processed.pop(k)
+            processed.pop(k, None)
         
         outputs = MODEL_STATE["model"].generate(
             **processed,
@@ -139,10 +139,11 @@ def ground_phrase(frontal_path, phrase):
             return_tensors="pt"
         ).to("cpu")
         
-        # Remove all image size related keys
+        # Convert to regular dict and remove image size related keys
+        processed = dict(processed)
         image_size_keys = [k for k in processed.keys() if "image_sizes" in k]
         for k in image_size_keys:
-            processed.pop(k)
+            processed.pop(k, None)
         
         outputs = MODEL_STATE["model"].generate(
             **processed,
@@ -235,4 +236,4 @@ with gr.Blocks(title="MAIRA-2 Medical Assistant") as demo:
                 outputs=pg_output
             )
 
-demo.launch()
+demo.launch()