Create app.py

app.py

@@ -0,0 +1,32 @@
+import gradio as gr
+from inference import Inference
+import os
+from huggingface_hub import snapshot_download
+
+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
+
+with gr.Blocks() as demo:
+    gr.Markdown("SuperRetina CFP Registration Demo")
+    
+    with gr.Row():
+        with gr.Column():
+            image_input = [gr.Image(type="numpy", label="Source Image"),
+                           gr.Image(type="numpy", label="Target Image")]
+        with gr.Column():
+            image_output = [gr.Image(type="numpy", label="Output"),
+                            gr.Image(type="numpy", label="Match Show")]
+    
+    # 当图像输入发生变化时自动触发推理
+    image_input.change(
+        fn=main,
+        inputs=image_input,
+        outputs=image_output
+    )
+
+demo.launch()