mehdiabruee commited on
Commit
3b7585f
1 Parent(s): fa92931

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ import torchvision
16
  import gradio as gr
17
 
18
  #dls = get_dls_from_hf("huggan/horse2zebra", load_size=286)
19
- cycle_gan = CycleGAN.from_pretrained('mehdiabruee/HW4')
20
 
21
  totensor = torchvision.transforms.ToTensor()
22
  normalize_fn = torchvision.transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
 
16
  import gradio as gr
17
 
18
  #dls = get_dls_from_hf("huggan/horse2zebra", load_size=286)
19
+ cycle_gan = CycleGAN.from_pretrained('./mehdiabruee/HW4')
20
 
21
  totensor = torchvision.transforms.ToTensor()
22
  normalize_fn = torchvision.transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))