Update LowLight.py

LowLight.py

@@ -19,7 +19,7 @@ def inference_img(img_path,Net):
     data_lowlight = (np.asarray(low_image)/255.0)
     data_lowlight = torch.from_numpy(data_lowlight).float()
     data_lowlight = data_lowlight.permute(2,0,1)
-    low_image = data_lowlight.cuda().unsqueeze(0)
+    low_image = data_lowlight.unsqueeze(0)
     
     with torch.no_grad():
         start = time.time()