Update app.py

app.py

@@ -1,3 +1,32 @@
+import os
+
+def replace_import_line(file_path):
+    # 置換対象の行
+    old_line = "from torchvision.transforms.functional_tensor import rgb_to_grayscale"
+    # 新しい行
+    new_line = "from torchvision.transforms.functional import rgb_to_grayscale"
+
+    # ファイルが存在するかチェック
+    if not os.path.exists(file_path):
+        print(f"ファイルが見つかりません: {file_path}")
+        return
+
+    # ファイルを読み込んで内容を置換
+    with open(file_path, 'r') as file:
+        lines = file.readlines()
+
+    # 置換処理
+    updated_lines = [line.replace(old_line, new_line) if old_line in line else line for line in lines]
+
+    # ファイルに書き込み
+    with open(file_path, 'w') as file:
+        file.writelines(updated_lines)
+
+    print(f"ファイルの置換が完了しました: {file_path}")
+
+# 書き換えるファイルのパスを指定
+file_path = "basicsr/data/degradations.py"
+replace_import_line(file_path)
 import gradio as gr
 import cv2
 import numpy as np