NameError: name 'model_fn' is not defined

[ShravanKumarChiluka]

Hello,
As per the instructions mentioned in the README.md document I tried to execute the project
When I try to execute the file named "demo_grayscale.py" and demo_real.py I'm getting the following error
NameError: name 'model_fn' is not defined for both the files.
The function "model_fn" is not defined and please help me to resolve this error as this will be a big help to proceed regarding my project.

Thanks.
Following is the screenshot of the error which I encountered.

[sanghviyashiitb]

Hi,

Apologies for the bug, I had not tested the code to run on a CPU. I think the code should run by commenting lines 90-93. The model weights are loaded into the network anyway.

However, I really advise you against running the code without a GPU since it would require a lot of time to just run a single deconvolution result.