feng2022 commited on
Commit
a9f4e65
1 Parent(s): b04b504

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -38,7 +38,7 @@ ARTICLE = '<center><img src="https://visitor-badge.glitch.me/badge?page_id=Time-
38
  TOKEN = "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"
39
 
40
  def image_create(seed: int, truncation_psi: float):
41
- args = ProjectorArguments().parse(
42
  args=[str(input_path)],
43
  namespace=Namespace(
44
  encoder_ckpt=f"checkpoint/encoder/checkpoint_{spectral_sensitivity}.pt",
@@ -50,8 +50,8 @@ def image_create(seed: int, truncation_psi: float):
50
  latent = torch.randn((1, 512), device=device)
51
  img_out, _, _ = generator([latent])
52
  imgs_arr = make_image(img_out)
53
- return imgs_arr[0][0]
54
- #return np.zeros((10,10,10))
55
 
56
  def main():
57
  #torch.cuda.init()
 
38
  TOKEN = "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"
39
 
40
  def image_create(seed: int, truncation_psi: float):
41
+ """args = ProjectorArguments().parse(
42
  args=[str(input_path)],
43
  namespace=Namespace(
44
  encoder_ckpt=f"checkpoint/encoder/checkpoint_{spectral_sensitivity}.pt",
 
50
  latent = torch.randn((1, 512), device=device)
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()