Update app.py

app.py

@@ -1,10 +1,7 @@
 import os
 import shutil
 import torch
-from natsort import natsorted
-from glob import glob
 import cv2
-import numpy as np
 import gradio as gr
 from PIL import Image
 
@@ -57,18 +54,18 @@ def inference(img,task):
 
     if task == 'Motion Deblurring':
       task = 'Motion_Deblurring'
-      os.system("python demo.py --task 'Motion_Deblurring' --input_dir './temp/image.jpg' --result_dir './temp/'")
+      os.system("python demo_gradio.py --task 'Motion_Deblurring' --input_dir './temp/image.jpg' --result_dir './temp/'")
   
     if task == 'Defocus Deblurring':
       task = 'Single_Image_Defocus_Deblurring'
-      os.system("python demo.py --task 'Single_Image_Defocus_Deblurring' --input_dir './temp/image.jpg' --result_dir './temp/'")
+      os.system("python demo_gradio.py --task 'Single_Image_Defocus_Deblurring' --input_dir './temp/image.jpg' --result_dir './temp/'")
   
     if task == 'Denoising':
       task = 'Real_Denoising'
-      os.system("python demo.py --task 'Real_Denoising' --input_dir './temp/image.jpg' --result_dir './temp/'")
+      os.system("python demo_gradio.py --task 'Real_Denoising' --input_dir './temp/image.jpg' --result_dir './temp/'")
   
     if task == 'Deraining':
-      os.system("python demo.py --task 'Deraining' --input_dir './temp/image.jpg' --result_dir './temp/'")
+      os.system("python demo_gradio.py --task 'Deraining' --input_dir './temp/image.jpg' --result_dir './temp/'")
   
     return f'temp/{task}/image.png'