FIX loading super resolution repo

app.py

@@ -1,4 +1,20 @@
 # %%
+import os
+
+from PIL import Image
+
+os.system('wget https://github.com/JingyunLiang/SwinIR/releases/download/v0.0/003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x4_GAN.pth -P experiments/pretrained_models')
+
+def super_resolution_inference(img):
+    os.system('mkdir test')
+    basewidth = 256
+    wpercent = (basewidth/float(img.size[0]))
+    hsize = int((float(img.size[1])*float(wpercent)))
+    img = img.resize((basewidth,hsize), Image.ANTIALIAS)
+    img.save("test/1.jpg", "JPEG")
+    os.system('python main_test_swinir.py --task real_sr --model_path experiments/pretrained_models/003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x4_GAN.pth --folder_lq test --scale 4')
+    return 'results/swinir_real_sr_x4/1_SwinIR.png'
+# %%
 import gradio as gr
 
 example_generated_ape_super = "examples/generated_ape_super_resolution.jpg"
@@ -13,7 +29,8 @@ examples = [
 ]
 
 ape_gen = gr.Interface.load("spaces/ykilcher/apes")
-super_resolution = gr.Interface.load("spaces/akhaliq/SwinIR")
+# super_resolution = gr.Interface.load("spaces/akhaliq/SwinIR")
+super_resolution = super_resolution_inference
 style_transfer = gr.Interface.load("spaces/aravinds1811/neural-style-transfer")
 
 def generate_ape(num_images=1, interpolate=False):
@@ -28,12 +45,12 @@ def perform_style_transfer(content_image, style_image, is_super_resolution=False
 # %%
 with gr.Blocks() as demo:
     button_generate_ape = gr.Button("Generate Ape")
-    
+
     generated_ape = gr.Image(example_generated_ape, label="Generated Ape", type="filepath")
     style_image_input = gr.Image(example_style_starry_night, label="Stylish Image", type="filepath")
     style_in_super_resolution = gr.Checkbox(True, label="Super resolution!")
     stylish_ape = gr.Image(example_stylish_ape, label="Stylish Ape")
-    
+
     gr.Interface(
         fn=perform_style_transfer,
         inputs=[generated_ape, style_image_input, style_in_super_resolution],