MaBuuMinh commited on
Commit
899a074
·
verified ·
1 Parent(s): 0bec222

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -11,6 +11,12 @@ os.system("pip install basicsr facexlib gfpgan ffmpeg-python ipython")
11
  os.system("pip install -r requirements.txt")
12
  os.system("python setup.py develop")
13
 
 
 
 
 
 
 
14
  from IPython.display import HTML
15
 
16
  # Set up folders
 
11
  os.system("pip install -r requirements.txt")
12
  os.system("python setup.py develop")
13
 
14
+ with open('/usr/local/lib/python3.10/site-packages/basicsr/data/degradations.py', 'r') as file:
15
+ file_data = file.read()
16
+ file_data = file_data.replace('from torchvision.transforms.functional_tensor import rgb_to_grayscale', 'from torchvision.transforms.functional import rgb_to_grayscale')
17
+ with open('/usr/local/lib/python3.10/site-packages/basicsr/data/degradations.py', 'w') as file:
18
+ file.write(file_data)
19
+
20
  from IPython.display import HTML
21
 
22
  # Set up folders