app

app.py

@@ -81,20 +81,18 @@ def sepia(image_web, seed):
     np.random.seed(seed)
     if torch.cuda.is_available():
         torch.cuda.manual_seed_all(seed)
-    image = web_input(image_web)
+    image, image_tmp = web_input(image_web)
     output_image = model.web_restore(image, r=config.data.grid_r)
     grid = make_grid(output_image)
     ndarr = grid.mul(255).add_(0.5).clamp_(0, 255).permute(1, 2, 0).to("cpu", torch.uint8).numpy()
-    # torch.cuda.empty_cache()
-    return ndarr
+    torch.cuda.empty_cache()
+    return ndarr, image_tmp
 
 
 demo = gr.Interface(sepia,
                     inputs=[gr.Image(label="输入甲骨文图片", height=600, width=600), gr.Number(label="随机种子")],
-                    outputs=gr.Image(label="输出汉字图片", height=600, width=600),
+                    outputs=[gr.Image(label="输出汉字图片", height=600, width=600), gr.Image(label="矫正后甲骨文图片", height=600, width=600)],
                     title=title_markdown)
 demo.queue().launch(
-    server_name="127.0.0.1",
-    server_port=7681,
     share=True
 )