Upgrade run_cmd

app.py

@@ -5,27 +5,37 @@ from PIL import Image
 import torch
 from random import randint
 import sys
-from subprocess import call
 import psutil
+import subprocess
+import sys
 
-
-
-
-torch.hub.download_url_to_file('http://people.csail.mit.edu/billf/project%20pages/sresCode/Markov%20Random%20Fields%20for%20Super-Resolution_files/100075_lowres.jpg', 'bear.jpg')
-  
-    
 def run_cmd(command):
     try:
-        print(command)
-        call(command, shell=True)
+        print(f"Running command: {command}")
+        # Run the command and capture both output and error
+        result = subprocess.run(command, shell=True, text=True, capture_output=True)
+
+        # Print stdout and stderr
+        if result.stdout:
+            print("Output:\n", result.stdout)
+        
+        if result.stderr:
+            print("Error:\n", result.stderr)
+        
+        # Check for command success
+        if result.returncode != 0:
+            print(f"Command failed with return code {result.returncode}")
+
     except KeyboardInterrupt:
         print("Process interrupted")
         sys.exit(1)
+
+
 run_cmd("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth -P .")
 run_cmd("pip install basicsr")
 run_cmd("pip freeze")
 
-os.system("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth -P .")
+run_cmd("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth -P .")
 
 
 def inference(img,mode):
@@ -44,7 +54,7 @@ def inference(img,mode):
     if mode == "base":
         run_cmd("python inference_realesrgan.py -n RealESRGAN_x4plus -i "+ INPUT_DIR + " -o " + OUTPUT_DIR)
     else:
-        os.system("python inference_realesrgan.py -n RealESRGAN_x4plus_anime_6B -i "+ INPUT_DIR + " -o " + OUTPUT_DIR)
+        run_cmd("python inference_realesrgan.py -n RealESRGAN_x4plus_anime_6B -i "+ INPUT_DIR + " -o " + OUTPUT_DIR)
     return os.path.join(OUTPUT_DIR, "1_out.png")