Update app.py

app.py

@@ -7,6 +7,24 @@ import glob
 import psutil
 import time
 
+def modify_degradations_py():
+    file_path = '/usr/local/lib/python3.10/site-packages/basicsr/data/degradations.py'
+    with open(file_path, 'r') as f:
+        lines = f.readlines()
+
+    # Find the line containing 'from torchvision.transforms.functional_tensor import rgb_to_grayscale'
+    for i, line in enumerate(lines):
+        if 'from torchvision.transforms.functional_tensor import rgb_to_grayscale' in line:
+            # Replace it with 'from torchvision.transforms.functional import rgb_to_grayscale'
+            lines[i] = 'from torchvision.transforms.functional import rgb_to_grayscale\n'
+            break
+
+    with open(file_path, 'w') as f:
+        f.writelines(lines)
+
+# Call the function to modify the file
+modify_degradations_py()
+
 # Page configuration
 st.set_page_config(
     page_title="Image Enhancer",