add alinged case

app.py

@@ -24,7 +24,7 @@ from basicsr.utils.download_util import load_file_from_url
 cfg_path = 'configs/sample/iddpm_ffhq512_swinir.yaml'
 configs = OmegaConf.load(cfg_path)
 configs.aligned = False
-configs.diffusion.timestep_respacing = '200'
+configs.diffusion.timestep_respacing = '250'
 
 # build the sampler for diffusion
 sampler_dist = DifIRSampler(configs)
@@ -92,62 +92,62 @@ def predict(im_path, background_enhance, face_upsample, upscale, started_timeste
 
     return image_restored, str(save_path)
 
-# title = "DifFace: Blind Face Restoration with Diffused Error Contraction"
-# description = r"""
-# <b>Official Gradio demo</b> for <a href='https://github.com/zsyOAOA/DifFace' target='_blank'><b>DifFace: Blind Face Restoration with Diffused Error Contraction</b></a>.<br>
-# 🔥 DifFace is a robust face restoration algorithm for old or corrupted photos.<br>
-# """
-# article = r"""
-# If DifFace is helpful for your work, please help to ⭐ the <a href='https://github.com/zsyOAOA/DifFace' target='_blank'>Github Repo</a>. Thanks!
-# [![GitHub Stars](https://img.shields.io/github/stars/zsyOAOA/DifFace?affiliations=OWNER&color=green&style=social)](https://github.com/zsyOAOA/DifFace)
-
-# ---
-
-# 📝 **Citation**
-
-# If our work is useful for your research, please consider citing:
-# ```bibtex
-# @article{yue2022difface,
-  # title={DifFace: Blind Face Restoration with Diffused Error Contraction},
-  # author={Yue, Zongsheng and Loy, Chen Change},
-  # journal={arXiv preprint arXiv:2212.06512},
-  # year={2022}
-# }
-# ```
-
-# 📋 **License**
-
-# This project is licensed under <a rel="license" href="https://github.com/zsyOAOA/DifFace/blob/master/LICENSE">S-Lab License 1.0</a>.
-# Redistribution and use for non-commercial purposes should follow this license.
-
-# 📧 **Contact**
-# If you have any questions, please feel free to contact me via <b>[email protected]</b>.
-# ![visitors](https://visitor-badge.laobi.icu/badge?page_id=zsyOAOA/DifFace)
-# """
-
-# demo = gr.Interface(
-    # predict,
-    # inputs=[
-        # gr.Image(type="filepath", label="Input"),
-        # gr.Checkbox(value=True, label="Background_Enhance"),
-        # gr.Checkbox(value=True, label="Face_Upsample"),
-        # gr.Number(value=2, label="Rescaling_Factor (up to 4)"),
-        # gr.Slider(1, 160, value=80, step=10, label='Realism-Fidelity Trade-off')
-    # ],
-    # outputs=[
-        # gr.Image(type="numpy", label="Output"),
-        # gr.outputs.File(label="Download the output")
-    # ],
-    # title=title,
-    # description=description,
-    # article=article,
-    # examples=[
-        # ['./testdata/whole_imgs/00.jpg', True, True, 2, 80],
-        # ['./testdata/whole_imgs/01.jpg', True, True, 2, 80],
-        # ['./testdata/whole_imgs/04.jpg', True, True, 2, 80],
-        # ['./testdata/whole_imgs/05.jpg', True, True, 2, 80],
-      # ]
-    # )
-
-# demo.queue(concurrency_count=4)
-# demo.launch()
+title = "DifFace: Blind Face Restoration with Diffused Error Contraction"
+description = r"""
+<b>Official Gradio demo</b> for <a href='https://github.com/zsyOAOA/DifFace' target='_blank'><b>DifFace: Blind Face Restoration with Diffused Error Contraction</b></a>.<br>
+🔥 DifFace is a robust face restoration algorithm for old or corrupted photos.<br>
+"""
+article = r"""
+If DifFace is helpful for your work, please help to ⭐ the <a href='https://github.com/zsyOAOA/DifFace' target='_blank'>Github Repo</a>. Thanks!
+[![GitHub Stars](https://img.shields.io/github/stars/zsyOAOA/DifFace?affiliations=OWNER&color=green&style=social)](https://github.com/zsyOAOA/DifFace)
+
+---
+
+📝 **Citation**
+
+If our work is useful for your research, please consider citing:
+```bibtex
+@article{yue2022difface,
+  title={DifFace: Blind Face Restoration with Diffused Error Contraction},
+  author={Yue, Zongsheng and Loy, Chen Change},
+  journal={arXiv preprint arXiv:2212.06512},
+  year={2022}
+}
+```
+
+📋 **License**
+
+This project is licensed under <a rel="license" href="https://github.com/zsyOAOA/DifFace/blob/master/LICENSE">S-Lab License 1.0</a>.
+Redistribution and use for non-commercial purposes should follow this license.
+
+📧 **Contact**
+If you have any questions, please feel free to contact me via <b>[email protected]</b>.
+![visitors](https://visitor-badge.laobi.icu/badge?page_id=zsyOAOA/DifFace)
+"""
+
+demo = gr.Interface(
+    predict,
+    inputs=[
+        gr.Image(type="filepath", label="Input"),
+        gr.Checkbox(value=True, label="Background_Enhance"),
+        gr.Checkbox(value=True, label="Face_Upsample"),
+        gr.Number(value=2, label="Rescaling_Factor (up to 4)"),
+        gr.Slider(1, 160, value=100, step=10, label='Realism-Fidelity Trade-off')
+    ],
+    outputs=[
+        gr.Image(type="numpy", label="Output"),
+        gr.outputs.File(label="Download the output")
+    ],
+    title=title,
+    description=description,
+    article=article,
+    examples=[
+        ['./testdata/whole_imgs/00.jpg', True, True, 2, 100],
+        ['./testdata/whole_imgs/01.jpg', True, True, 2, 100],
+        ['./testdata/whole_imgs/04.jpg', True, True, 2, 100],
+        ['./testdata/whole_imgs/05.jpg', True, True, 2, 100],
+      ]
+    )
+
+demo.queue(concurrency_count=4)
+demo.launch()