guetLzy commited on
Commit
0c9b170
·
verified ·
1 Parent(s): 685562e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -3,7 +3,14 @@ import cv2
3
  import subprocess
4
 
5
  import subprocess
6
- # subprocess.run(["sed", "-i", "9s/.*/import torchvision.transforms.functional as F_t/", "/usr/local/lib/python3.10/dist-packages/pytorchvideo/transforms/augmentations.py"], check=True)
 
 
 
 
 
 
 
7
 
8
  import torch
9
  from basicsr.archs.rrdbnet_arch import RRDBNet
 
3
  import subprocess
4
 
5
  import subprocess
6
+ import subprocess
7
+
8
+ # 修复导入语句
9
+ subprocess.run([
10
+ "sed", "-i",
11
+ "8s/from torchvision.transforms.functional_tensor import rgb_to_grayscale/from torchvision.transforms.functional import rgb_to_grayscale/",
12
+ "/usr/local/lib/python3.10/site-packages/basicsr/data/degradations.py"
13
+ ], check=True)
14
 
15
  import torch
16
  from basicsr.archs.rrdbnet_arch import RRDBNet