Cropinky commited on
Commit
d232680
1 Parent(s): 24715f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -37,7 +37,6 @@ def image_generation(model, number_of_images=1):
37
  tile_pad=10,
38
  pre_pad=0,
39
  half=False,
40
- device='cpu'
41
  )
42
  #TODO: img has to be same as opencv imread format
43
  open_cv_image = np.array(img)
37
  tile_pad=10,
38
  pre_pad=0,
39
  half=False,
 
40
  )
41
  #TODO: img has to be same as opencv imread format
42
  open_cv_image = np.array(img)