Update app.py

app.py

@@ -45,14 +45,12 @@ def image_create(seed: int, truncation_psi: float):
             #gaussian=gaussian_radius,
             log_visual_freq=1000
     ))
-    #device = th.device()
-    #generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args,device)
-    """latent = torch.randn((1, 512), device=device) 
+    device = th.device()
+    generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args,device)
+    latent = torch.randn((1, 512), device=device) 
     img_out, _, _ = generator([latent])
     imgs_arr = make_image(img_out)
-    return imgs_arr[0][0]"""
-    n=np.zeros((10,10,3))
-    n[0][0]=[0,0,254]
+    return imgs_arr[0]/255
     return n
     
 def main():