fix pylint

TencentARC · Sep 16, 2022 · 2f46d95 · 2f46d95
1 parent bc5a5de
commit 2f46d95
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/cog_predict.py b/cog_predict.py
@@ -64,13 +64,13 @@ def setup(self):
         self.current_version = 'v1.4'
 
     def predict(
-        self,
-        img: Path = Input(description='Input'),
-        version: str = Input(
-            description='GFPGAN version. v1.3: better quality. v1.4: more details and better identity.',
-            choices=['v1.2', 'v1.3', 'v1.4', 'RestoreFormer'],
-            default='v1.4'),
-        scale: float = Input(description='Rescaling factor', default=2),
+            self,
+            img: Path = Input(description='Input'),
+            version: str = Input(
+                description='GFPGAN version. v1.3: better quality. v1.4: more details and better identity.',
+                choices=['v1.2', 'v1.3', 'v1.4', 'RestoreFormer'],
+                default='v1.4'),
+            scale: float = Input(description='Rescaling factor', default=2),
     ) -> Path:
         weight = 0.5
         print(img, version, scale, weight)