Cleanup

app.py

@@ -13,8 +13,6 @@ import psutil
 
 
 
-
-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):
@@ -24,11 +22,13 @@ 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("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.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):
@@ -39,10 +39,7 @@ def inference(img,mode):
     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("python inference_realesrgan.py -n RealESRGAN_x4plus -i "+ INPUT_DIR + " -o " + OUTPUT_DIR)
@@ -59,12 +56,9 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2107.108
 
 gr.Interface(
     inference, 
-    [gr.inputs.Image(type="pil", label="Input"),gr.inputs.Radio(["base","anime"], type="value", default="base", label="model type")], 
+    [gr.inputs.Image(type="pil", label="Input"),gr.inputs.Radio(["base","anime"], type="value", default="anime", label="model type")], 
     gr.outputs.Image(type="file", label="Output"),
     title=title,
     description=description,
-    article=article,
-    examples=[
-    #['bear.jpg','base'],
-    #['anime.png','anime']
-    ]).launch()
+    article=article
+).launch()