Update app.py

app.py

@@ -24,10 +24,10 @@ run_cmd("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/Re
 run_cmd("pip install basicsr")
 run_cmd("pip freeze")
 
-#run_cmd("python setup.py develop")
+os.system("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth -P .")
 
 
-def inference(img):
+def inference(img,mode):
     _id = randint(1, 10000)
     INPUT_DIR = "/tmp/input_image" + str(_id) + "/"
     OUTPUT_DIR = "/tmp/output_image" + str(_id) + "/"
@@ -40,7 +40,10 @@ def inference(img):
     hsize = int((float(img.size[1])*float(wpercent)))
     img = img.resize((basewidth,hsize), Image.ANTIALIAS)
     img.save(INPUT_DIR + "1.jpg", "JPEG")
-    run_cmd("python inference_realesrgan.py --model_path RealESRGAN_x4plus.pth --input "+ INPUT_DIR + " --output " + OUTPUT_DIR + " --netscale 4 --outscale 4")
+    if mode == "base":
+        run_cmd("python inference_realesrgan.py --model_path RealESRGAN_x4plus.pth --input "+ INPUT_DIR + " --output " + OUTPUT_DIR + " --netscale 4 --outscale 4")
+    else:
+        os.system("python inference_realesrgan.py -n RealESRGAN_x4plus_anime_6B -i "+ INPUT_DIR + " --output " + OUTPUT_DIR)
     return os.path.join(OUTPUT_DIR, "1_out.jpg")
 
 
@@ -52,7 +55,7 @@ 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.Image(type="pil", label="Input"),gradio.inputs.Radio(["base","anime"], type="value", default="base", label="model type")], 
     gr.outputs.Image(type="file", label="Output"),
     title=title,
     description=description,