swzamir commited on
Commit
dad1674
1 Parent(s): 7397a8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -54,7 +54,7 @@ def inference(img,task):
54
  img = img.resize((width,height), Image.ANTIALIAS)
55
 
56
  img.save("temp/image.jpg", "JPEG")
57
- """
58
  if task == 'Motion Deblurring':
59
  task = 'Motion_Deblurring'
60
  os.system("python demo.py --task 'Motion_Deblurring' --input_dir './temp/image.jpg' --result_dir './temp/'")
@@ -70,11 +70,7 @@ def inference(img,task):
70
  if task == 'Deraining':
71
  os.system("python demo.py --task 'Deraining' --input_dir './temp/image.jpg' --result_dir './temp/'")
72
 
73
-
74
- restored = f'temp/{task}/image.png'
75
- """
76
- restored = 'temp/image.jpg'
77
- return restored
78
 
79
  gr.Interface(
80
  inference,
 
54
  img = img.resize((width,height), Image.ANTIALIAS)
55
 
56
  img.save("temp/image.jpg", "JPEG")
57
+
58
  if task == 'Motion Deblurring':
59
  task = 'Motion_Deblurring'
60
  os.system("python demo.py --task 'Motion_Deblurring' --input_dir './temp/image.jpg' --result_dir './temp/'")
 
70
  if task == 'Deraining':
71
  os.system("python demo.py --task 'Deraining' --input_dir './temp/image.jpg' --result_dir './temp/'")
72
 
73
+ return f'temp/{task}/image.png'
 
 
 
 
74
 
75
  gr.Interface(
76
  inference,