Hugging Face's logo

Spaces:
luismidv
/
NeuralStyleTransfer
Sleeping

App Files Community
luismidv commited on
Commit
930f9e5
·
1 Parent(s): ca7ff3b

new

Browse files
Files changed (1) hide show
  1. train.py +2 -2
train.py CHANGED
@@ -24,8 +24,8 @@ def white_noise_img(model, original_image, content_image, epochs = 5):
24
  print(f"Loss at content type: {content_loss}")
25
 
26
 
27
- image = dataprepare.image_process('/content/drive/MyDrive/16546923557574.jpg')
28
- content_image = dataprepare.image_process('/content/drive/MyDrive/imagen2.png')
29
  print(image.shape)
30
  print(content_image.shape)
31
  image = image.view(1, image.shape[0],image.shape[1],image.shape[2])
 
24
  print(f"Loss at content type: {content_loss}")
25
 
26
 
27
+ image = dataprepare.image_process('/data/16546923557574.jpg')
28
+ content_image = dataprepare.image_process('/data/imagen2.png')
29
  print(image.shape)
30
  print(content_image.shape)
31
  image = image.view(1, image.shape[0],image.shape[1],image.shape[2])