修复

app.py

@@ -1,8 +1,8 @@
 '''
 Author: Egrt
 Date: 2022-01-13 13:34:10
-LastEditors: Egrt
-LastEditTime: 2022-01-13 13:48:57
+LastEditors: [egrt]
+LastEditTime: 2022-04-08 21:22:10
 FilePath: \LicenseGAN\app.py
 '''
 import os
@@ -27,7 +27,7 @@ description = "使用生成对抗网络对低分辨率车牌图片进行八倍
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2108.10257' target='_blank'>LicenseGAN: Image Restoration Using Swin Transformer</a> | <a href='https://github.com/JingyunLiang/SwinIR' target='_blank'>Github Repo</a></p>"
 example_img_dir  = 'img'
 example_img_name = os.listdir(example_img_dir)
-examples=[[os.path.join(example_img_dir, image_path)] for image_path in example_img_name if image_path.endswith('.jpg')]
+examples=[[os.path.join(example_img_dir, image_path)] for image_path in example_img_name if image_path.endswith('.jpg', '.png', '.tif')]
 gr.Interface(
     inference, 
     [gr.inputs.Image(type="pil", label="Input")],