Update app.py

app.py

@@ -5,27 +5,18 @@ import cv2
 import gradio as gr
 from PIL import Image
 
-os.system("pip install einops")
-os.system("git clone https://github.com/swz30/Restormer.git")
-shutil.copy("demo_gradio.py","Restormer/demo_gradio.py")
 os.chdir('Restormer')
 
-# Download pretrained models
-os.system("wget https://github.com/swz30/Restormer/releases/download/v1.0/real_denoising.pth -P Denoising/pretrained_models")
-os.system("wget https://github.com/swz30/Restormer/releases/download/v1.0/single_image_defocus_deblurring.pth -P Defocus_Deblurring/pretrained_models")
-os.system("wget https://github.com/swz30/Restormer/releases/download/v1.0/motion_deblurring.pth -P Motion_Deblurring/pretrained_models")
-os.system("wget https://github.com/swz30/Restormer/releases/download/v1.0/deraining.pth -P Deraining/pretrained_models")
-
 # Download sample images
-os.system("wget https://github.com/swz30/Restormer/releases/download/v1.0/sample_images.zip -P demo")
-shutil.unpack_archive('demo/sample_images.zip', 'demo/')
-os.remove('demo/sample_images.zip')
+os.system("wget https://github.com/swz30/Restormer/releases/download/v1.0/sample_images.zip")
+shutil.unpack_archive('sample_images.zip')
+os.remove('sample_images.zip')
 
 
-examples = [['demo/sample_images/Real_Denoising/degraded/117355.png', 'Denoising'],
-            ['demo/sample_images/Single_Image_Defocus_Deblurring/degraded/engagement.jpg', 'Defocus Deblurring'],
-            ['demo/sample_images/Motion_Deblurring/degraded/GoPro-GOPR0854_11_00-000090-input.jpg','Motion Deblurring'],
-            ['demo/sample_images/Deraining/degraded/Rain100H-77-input.jpg','Deraining']]
+examples = [['sample_images/Real_Denoising/degraded/117355.png', 'Denoising'],
+            ['sample_images/Single_Image_Defocus_Deblurring/degraded/engagement.jpg', 'Defocus Deblurring'],
+            ['sample_images/Motion_Deblurring/degraded/GoPro-GOPR0854_11_00-000090-input.jpg','Motion Deblurring'],
+            ['sample_images/Deraining/degraded/Rain100H-77-input.jpg','Deraining']]
 
 
 title = "Restormer"