Update app.py

app.py

@@ -38,7 +38,7 @@ ARTICLE = '<center><img src="https://visitor-badge.glitch.me/badge?page_id=Time-
 TOKEN = "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"
    
 def image_create(seed: int, truncation_psi: float):
-    """args = ProjectorArguments().parse(
+    args = ProjectorArguments().parse(
         args=[str(input_path)],
         namespace=Namespace(
             encoder_ckpt=f"checkpoint/encoder/checkpoint_{spectral_sensitivity}.pt",
@@ -50,8 +50,8 @@ def image_create(seed: int, truncation_psi: float):
     latent = torch.randn((1, 512), device=device) 
     img_out, _, _ = generator([latent])
     imgs_arr = make_image(img_out)
-    return imgs_arr[0].cpu().numpy()"""
-    return np.zeros((10,10))
+    return imgs_arr[0][0]
+    #return np.zeros((10,10))
     
 def main():
     #torch.cuda.init()