Tas112 commited on
Commit
417deab
·
verified ·
1 Parent(s): 6dedb66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -6
app.py CHANGED
@@ -24,12 +24,16 @@ import requests
24
 
25
  try:
26
  from realesrgan import RealESRGAN
27
- except ImportError:
28
- import subprocess
29
- subprocess.run([sys.executable, "-m", "pip", "install", "torchvision==0.16.0"], check=True)
30
- subprocess.run(["git", "clone", "https://github.com/xinntao/Real-ESRGAN.git", "/tmp/Real-ESRGAN"], check=True)
31
- subprocess.run([sys.executable, "-m", "pip", "install", "-r", "/tmp/Real-ESRGAN/requirements.txt"], check=True)
32
- subprocess.run([sys.executable, "setup.py", "develop"], cwd="/tmp/Real-ESRGAN", check=True)
 
 
 
 
33
  from realesrgan import RealESRGAN
34
 
35
  def download_file(url, folder_path, filename):
 
24
 
25
  try:
26
  from realesrgan import RealESRGAN
27
+ except (ImportError, ModuleNotFoundError):
28
+ degradations_file = "/usr/local/lib/python3.10/site-packages/basicsr/data/degradations.py"
29
+ with open(degradations_file, "r") as f:
30
+ content = f.read()
31
+ content = content.replace(
32
+ "from torchvision.transforms.functional_tensor import rgb_to_grayscale",
33
+ "from torchvision.transforms.functional import rgb_to_grayscale"
34
+ )
35
+ with open(degradations_file, "w") as f:
36
+ f.write(content)
37
  from realesrgan import RealESRGAN
38
 
39
  def download_file(url, folder_path, filename):