model was corrupted

.gitignore

@@ -64,4 +64,7 @@ venv.bak/
 .vscode/
 
 # tmp folder
-.tmp
+.tmp
+
+# gradio folder
+.gradio

app.py

@@ -366,6 +366,8 @@ class LocalizationModel:
 
 
 def infer(image1, image2):
+    assert isinstance(image1, Image.Image), "image1 is not a PIL Image"
+    assert isinstance(image2, Image.Image), "image2 is not a PIL Image"
     localization_model = LocalizationModel()
     change_detection_model = ChangeDetectionModel()
     # resize image1 to image2
@@ -410,6 +412,11 @@ sample_images = [
     ["assets/data/beirut_1_pre.png", "assets/data/beirut_1_post.png"]
 ]
 
+# Check if all files exist
+for pair in sample_images:
+    for file in pair:
+        assert os.path.exists(file), f"File not found: {file}"
+        
 with gr.Blocks() as demo:
     gr.Markdown("## Infrastructure Damage Assessment")
     # description
@@ -437,9 +444,14 @@ with gr.Blocks() as demo:
         output_image1 = gr.Image(label="Roads and buildings localization", type="pil")
         output_image2 = gr.Image(label="Change detection", type="pil")
     submit_button = gr.Button("Run Inference")
-    examples = gr.Examples(examples=sample_images, inputs=[
-                           input_image1, input_image2])
-    submit_button.click(fn=infer, inputs=[input_image1, input_image2], outputs=[
-                        output_image1, output_image2])
+    examples = gr.Examples(
+        examples=sample_images, 
+        inputs=[input_image1, input_image2]
+    )
+    submit_button.click(
+        infer,
+        inputs=[input_image1, input_image2],
+        outputs=[output_image1, output_image2],
+    )
 
 demo.launch()

assets/models/localization.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:545413e05e89b0b3382b4aade9e8f2a5f8c825d5c4e91b77b6c4f57401ff3086
-size 295
+oid sha256:4fccf7582ce357e67b6f16be6c969a394a47fb11a62a311c99fa0fd7e529af1c
+size 361226794

requirements.txt

@@ -4,4 +4,5 @@ onnxruntime==1.21.0
 pandas==2.2.3
 pillow==11.1.0
 scipy==1.15.2
-tqdm==4.67.1
+tqdm==4.67.1
+pydantic==2.10.6