Update app.py

app.py

@@ -65,18 +65,18 @@ def inference(img, version, scale):
         if h < 300:
             img = cv2.resize(img, (w * 2, h * 2), interpolation=cv2.INTER_LANCZOS4)
 
-        if version == 'v1.2':
-            face_enhancer = GFPGANer(
-            model_path='GFPGANv1.2.pth', upscale=2, arch='clean', channel_multiplier=2, bg_upsampler=upsampler)
+       # if version == 'v1.2':
+         ##   face_enhancer = GFPGANer(
+          #  model_path='GFPGANv1.2.pth', upscale=2, arch='clean', channel_multiplier=2, bg_upsampler=upsampler)
         elif version == 'v1.3':
             face_enhancer = GFPGANer(
             model_path='GFPGANv1.3.pth', upscale=2, arch='clean', channel_multiplier=2, bg_upsampler=upsampler)
-        elif version == 'v1.4':
-            face_enhancer = GFPGANer(
-            model_path='GFPGANv1.4.pth', upscale=2, arch='clean', channel_multiplier=2, bg_upsampler=upsampler)
-        elif version == 'RestoreFormer':
-            face_enhancer = GFPGANer(
-            model_path='RestoreFormer.pth', upscale=2, arch='RestoreFormer', channel_multiplier=2, bg_upsampler=upsampler)
+       # elif version == 'v1.4':
+       #     face_enhancer = GFPGANer(
+       #     model_path='GFPGANv1.4.pth', upscale=2, arch='clean', channel_multiplier=2, bg_upsampler=upsampler)
+       # elif version == 'RestoreFormer':
+       #     face_enhancer = GFPGANer(
+        #    model_path='RestoreFormer.pth', upscale=2, arch='RestoreFormer', channel_multiplier=2, bg_upsampler=upsampler)
         # elif version == 'CodeFormer':
         #     face_enhancer = GFPGANer(
         #     model_path='CodeFormer.pth', upscale=2, arch='CodeFormer', channel_multiplier=2, bg_upsampler=upsampler)
@@ -126,7 +126,7 @@ demo = gr.Interface(
     inference, [
         gr.inputs.Image(type="filepath", label="Input"),
         # gr.inputs.Radio(['v1.2', 'v1.3', 'v1.4', 'RestoreFormer', 'CodeFormer'], type="value", default='v1.4', label='version'),
-        gr.inputs.Radio(['v1.2', 'v1.3', 'v1.4', 'RestoreFormer'], type="value", default='v1.4', label='version'),
+        gr.inputs.Radio(['v1.4'], type="value", default='v1.4', label='version'),
         gr.inputs.Number(label="Rescaling factor", default=2),
         # gr.Slider(0, 100, label='Weight, only for CodeFormer. 0 for better quality, 100 for better identity', default=50)
     ], [