Update app.py

app.py

@@ -11,14 +11,23 @@ os.system(
 @spaces.GPU
 def inference(img):
     os.makedirs("./test", exist_ok=True)
+    os.makedirs("./results", exist_ok=True)
     basewidth = 512
     wpercent = (basewidth / float(img.size[0]))
     hsize = int((float(img.size[1]) * float(wpercent)))
     img = img.resize((basewidth, hsize), Image.BILINEAR)
     img.save("test/1.png", "PNG")
-    os.system(
+    
+    result = os.system(
         'python main_test_CMFNet.py --input_dir test --weights experiments/pretrained_models/deblur_GoPro_CMFNet.pth')
-    return 'results/1.png'
+    
+    output_path = 'results/1.png'
+    if os.path.exists(output_path):
+        return output_path
+    else:
+        print(f"Error: Output file not found at {output_path}")
+        print(f"Command exit code: {result}")
+        return None
 
 
 title = "Compound Multi-branch Feature Fusion for Image Restoration (Deblur)"