Update app.py

app.py

@@ -67,11 +67,12 @@ def load_model(file_name: str, path:str,device: torch.device) -> nn.Module:
         model(z, label, force_fp32=True)
     return model
    
-#def image_create(input_img):
-    
-
-def greet(name):
-    return "Hello " + name + "!"
+def image_create(input_img):
+    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)
     
 def main():
     #torch.cuda.init()
@@ -91,30 +92,22 @@ def main():
             log_visual_freq=1000,
             input='text',
         ))
-    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)
-    #iface = gr.Interface(
-      #fn=image_create, 
-      #inputs='text',
-      #gr.outputs.Image(type='numpy', label='Output'),
-      #title=TITLE,
-      #description=DESCRIPTION,
-      #article=ARTICLE,
+    iface = gr.Interface(
+      fn=image_create, 
+      inputs='text',
+      gr.outputs.Image(type='numpy', label='Output'),
+      title=TITLE,
+      description=DESCRIPTION,
+      article=ARTICLE,
         #theme=args.theme,
         #allow_flagging=args.allow_flagging,
         #live=args.live,
-    #)
+    )
     
-    #iface.launch(
+    iface.launch()
         #enable_queue=args.enable_queue,
         #server_port=args.port,
-        #share=args.share,
-    #)
-    demo = gr.Interface(fn=image_create, inputs="text", outputs="image")
-    demo.launch()
+        #share=args.share,)
 if __name__ == '__main__':
     main()