Update app.py

app.py

@@ -8,8 +8,13 @@ model_path = snapshot_download(repo_id='Cangshanqingshi/SuperRetinaDemo')
 inference_engine = Inference(model_path=model_path)
 
 def main(source_image, target_image):
-    merged, match_show = inference_engine.inference(source_image, target_image)
-    return merged, match_show
+    # 只有当两个图像都存在时才执行推理
+    if source_image is not None and target_image is not None:
+        merged, match_show = inference_engine.inference(source_image, target_image)
+        return merged, match_show
+    else:
+        # 如果任何一个图像为空，则返回空结果
+        return None, None
 
 with gr.Blocks() as demo:
     gr.Markdown("SuperRetina CFP Registration Demo")
@@ -22,7 +27,7 @@ with gr.Blocks() as demo:
             output_image = gr.Image(type="numpy", label="Output")
             match_show_image = gr.Image(type="numpy", label="Match Show")
     
-    # 当任一图像输入发生变化时触发推理
+    # 当任一图像输入发生变化时触发推理（但实际推理只在两个图像都存在时执行）
     source_image_input.change(
         fn=main,
         inputs=[source_image_input, target_image_input],