Attempt to revert partially

app.py

@@ -6,12 +6,14 @@ import torch
 from random import randint
 import sys
 from subprocess import call
+import psutil
+
 
-# Updating Gradio to a more current version
-# os.system("pip install gradio")
 
-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')
 
+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)
@@ -19,33 +21,35 @@ def run_cmd(command):
     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 .")
 
-def inference(img, mode):
+
+def inference(img,mode):
     _id = randint(1, 10000)
-    INPUT_DIR = f"./tmp/input_image{_id}/"
-    OUTPUT_DIR = f"./tmp/output_image{_id}/"
-    
-    run_cmd(f"rm -rf {INPUT_DIR}")
-    run_cmd(f"rm -rf {OUTPUT_DIR}")
-    run_cmd(f"mkdir -p {INPUT_DIR}")
-    run_cmd(f"mkdir -p {OUTPUT_DIR}")
-    
-    img.save(os.path.join(INPUT_DIR, "1.png"), "PNG")
-    
-    # Use the appropriate command based on the selected model
+    INPUT_DIR = "./tmp/input_image" + str(_id) + "/"
+    OUTPUT_DIR = "./tmp/output_image" + str(_id) + "/"
+    run_cmd("rm -rf " + INPUT_DIR)
+    run_cmd("rm -rf " + OUTPUT_DIR)
+    run_cmd("mkdir -p " + INPUT_DIR)
+    run_cmd("mkdir -p " + OUTPUT_DIR)
+    basewidth = 256
+    wpercent = (basewidth/float(img.size[0]))
+    hsize = int((float(img.size[1])*float(wpercent)))
+    #img = img.resize((basewidth,hsize), Image.LANCZOS)
+    img.save(INPUT_DIR + "1.png", "PNG")
     if mode == "base":
-        run_cmd(f"python inference_realesrgan.py -n RealESRGAN_x4plus -i {INPUT_DIR} -o {OUTPUT_DIR}")
+        run_cmd("python inference_realesrgan.py -n RealESRGAN_x4plus -i "+ INPUT_DIR + " -o " + OUTPUT_DIR)
     else:
-        os.system(f"python inference_realesrgan.py -n RealESRGAN_x4plus_anime_6B -i {INPUT_DIR} -o {OUTPUT_DIR}")
-    
+        os.system("python inference_realesrgan.py -n RealESRGAN_x4plus_anime_6B -i "+ INPUT_DIR + " -o " + OUTPUT_DIR)
     return os.path.join(OUTPUT_DIR, "1_out.png")
 
+
+
+        
 title = "Real-ESRGAN"
 description = "Gradio demo for Real-ESRGAN. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below. Please click submit only once"
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2107.10833'>Real-ESRGAN: Training Real-World Blind Super-Resolution with Pure Synthetic Data</a> | <a href='https://github.com/xinntao/Real-ESRGAN'>Github Repo</a></p>"