Hugging Face's logo

Spaces:
GRMenon
/
ClarityGAN
Running

App Files Community
GRMenon commited on
Commit
4df709d
•
1 Parent(s): 14cee3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from torchvision import transforms
8
  import gradio as gr
9
 
10
  gan = CycleGenerator(num_residuals=6)
11
- gan.load_state_dict(torch.load("genC.pth.tar", map_location=torch.device('cpu')))
12
 
13
 
14
  def dehaze(img):
 
8
  import gradio as gr
9
 
10
  gan = CycleGenerator(num_residuals=6)
11
+ gan.load_state_dict(torch.load("genC.pth.tar", map_location=torch.device('cpu'))["model"])
12
 
13
 
14
  def dehaze(img):