Update app.py

app.py

@@ -1,7 +1,33 @@
 import os
+import sys
+import types
+
 import cv2
-import gradio as gr
 import torch
+import torchvision.transforms as _tv_transforms
+
+# ---------------------------------------------------------------------------
+# torchvision compatibility shim
+#
+# basicsr (pulled in by gfpgan / realesrgan) does:
+#     from torchvision.transforms.functional_tensor import rgb_to_grayscale
+# torchvision removed the `functional_tensor` module in 0.17 (Jan 2024). Since
+# ZeroGPU now requires torch >= 2.8 (=> torchvision >= 0.23), that import would
+# raise ModuleNotFoundError and crash the Space at startup. We register a tiny
+# stand-in module that forwards rgb_to_grayscale, BEFORE basicsr is imported.
+# ---------------------------------------------------------------------------
+try:
+    from torchvision.transforms.functional import rgb_to_grayscale as _rgb_to_grayscale
+except Exception:  # very new torchvision may only expose it under v2
+    from torchvision.transforms.v2.functional import rgb_to_grayscale as _rgb_to_grayscale
+
+if 'torchvision.transforms.functional_tensor' not in sys.modules:
+    _shim = types.ModuleType('torchvision.transforms.functional_tensor')
+    _shim.rgb_to_grayscale = _rgb_to_grayscale
+    sys.modules['torchvision.transforms.functional_tensor'] = _shim
+    setattr(_tv_transforms, 'functional_tensor', _shim)
+
+import gradio as gr
 from basicsr.archs.srvgg_arch import SRVGGNetCompact
 from gfpgan.utils import GFPGANer
 from realesrgan.utils import RealESRGANer
@@ -93,7 +119,7 @@ def inference_internal(img, version, scale):
         if h > 3500 or w > 3500:
             print('too large size')
             return None, None
-        
+
         if h < 300:
             img = cv2.resize(img, (w * 2, h * 2), interpolation=cv2.INTER_LANCZOS4)
 
@@ -150,13 +176,13 @@ title = "GFPGAN: Practical Face Restoration Algorithm"
 description = r"""Gradio demo Fix for <a href='https://github.com/TencentARC/GFPGAN' target='_blank'><b>GFPGAN: Towards Real-World Blind Face Restoration with Generative Facial Prior</b></a>.<br>
 It can be used to restore your **old photos** or improve **AI-generated faces**.<br>
 To use it, simply upload your image.<br>
-If GFPGAN is helpful, please help to ⭐ the <a href='https://github.com/Nick088Official/GFPGAN-Fix' target='_blank'>Github Repo</a> and recommend it to your friends 😊
+If GFPGAN is helpful, please help to star the <a href='https://github.com/Nick088Official/GFPGAN-Fix' target='_blank'>Github Repo</a> and recommend it to your friends.
 """
 article = r"""
 [![download](https://img.shields.io/github/downloads/TencentARC/GFPGAN/total.svg)](https://github.com/TencentARC/GFPGAN/releases)
 [![GitHub Stars](https://img.shields.io/github/stars/TencentARC/GFPGAN?style=social)](https://github.com/TencentARC/GFPGAN)
 [![arXiv](https://img.shields.io/badge/arXiv-Paper-<COLOR>.svg)](https://arxiv.org/abs/2101.04061)
-If you have any question, please email 📧 `xintao.wang@outlook.com` or `xintaowang@tencent.com` (original creators).
+If you have any question, please email `xintao.wang@outlook.com` or `xintaowang@tencent.com` (original creators).
 """
 
 # Only offer examples for images that actually downloaded.