Hugging Face's logo

Spaces:
molher
/
TorchFakes
Sleeping

App Files Community
molher commited on
Commit
ca301d8
·
1 Parent(s): a63d77e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ import os
6
  import numpy as np
7
  from PIL import Image
8
  import matplotlib.pyplot as plt
 
9
 
10
  DEVICE = 'cuda:0' if torch.cuda.is_available() else 'cpu'
11
  print(f'Running on device: {DEVICE.upper()}')
 
6
  import numpy as np
7
  from PIL import Image
8
  import matplotlib.pyplot as plt
9
+ import pickle
10
 
11
  DEVICE = 'cuda:0' if torch.cuda.is_available() else 'cpu'
12
  print(f'Running on device: {DEVICE.upper()}')