feng2022 commited on
Commit
82dca1d
·
1 Parent(s): bb28f4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -51,7 +51,9 @@ def image_create(seed: int, truncation_psi: float):
51
  img_out, _, _ = generator([latent])
52
  imgs_arr = make_image(img_out)
53
  return imgs_arr[0][0]"""
54
- return np.zeros((10,10,3))
 
 
55
 
56
  def main():
57
  #torch.cuda.init()
 
51
  img_out, _, _ = generator([latent])
52
  imgs_arr = make_image(img_out)
53
  return imgs_arr[0][0]"""
54
+ n=np.zeros((10,10,3)
55
+ n[0][0][0]=255
56
+ return n
57
 
58
  def main():
59
  #torch.cuda.init()