Update app.py

app.py

@@ -24,11 +24,16 @@ except Exception:
     rembg_remove = None
 
 # face restore
+# ---- Optional face restore (safe import) ----
+_HAS_GFP = False
 try:
-    from gfpgan import GFPGANer
-    _HAS_GFP = True
+    import gfpgan
+    if hasattr(gfpgan, "GFPGANer"):
+        GFPGANer = gfpgan.GFPGANer  # type: ignore
+        _HAS_GFP = True
 except Exception:
-    _HAS_GFP = False
+    GFPGANer = None  # type: ignore
+GFP = None
 
 # realesrgan (fallback upscaler)
 try:
@@ -105,7 +110,6 @@ SCHEDULERS = {
 PIPE_CACHE: Dict[str, object] = {}
 CONTROL_CACHE: Dict[str, ControlNetModel] = {}
 UPSCALE_PIPE: Optional[StableDiffusionUpscalePipeline] = None
-GFP: Optional[GFPGANer] = None
 REALSR: Optional[RealESRGAN] = None
 
 def set_sched(pipe, name: str):