Update app.py

app.py

@@ -46,8 +46,9 @@ def lowlight(image):
 	if not os.path.exists(image_path.replace('/'+image_path.split("/")[-1],'')):
 		os.makedirs(image_path.replace('/'+image_path.split("/")[-1],''))
 
-	grid = torchvision.utils.make_grid(enhanced_image)
-	return grid
+	torchvision.utils.save_image(enhanced_image, f'01.jpg')
+
+	return '01.jpg'
 
 
 title = "Compound Multi-branch Feature Fusion for Image Restoration (Deblur)"