Update app.py

app.py

@@ -31,7 +31,7 @@ def lowlight(image_path):
 	data_lowlight = data_lowlight.cpu().unsqueeze(0)
 
 	DCE_net = model.enhance_net_nopool().cpu()
-	DCE_net.load_state_dict(torch.load('snapshots/Epoch99.pth'))
+	DCE_net.load_state_dict(torch.load('Epoch99.pth'))
 	start = time.time()
 	_,enhanced_image,_ = DCE_net(data_lowlight)
 
@@ -65,7 +65,7 @@ article = "<p style='text-align: center'><a href='https://' target='_blank'>Comp
 
 examples = [['data/test_data//01.jpg'], ['data/test_data//02.jpg'], ['data/test_data//03.jpg'],]
 gr.Interface(
-    inference,
+    lowlight,
     [gr.inputs.Image(type="pil", label="Input")],
     gr.outputs.Image(type="file", label="Output"),
     title=title,