Update app.py

app.py

@@ -66,7 +66,7 @@ def image_create(input_img):
     latent = torch.randn((1, 512), device=device) 
     img_out, _, _ = generator([latent])
     imgs_arr = make_image(img_out)
-    return imgs_arr[0]
+    return imgs_arr[0].cpu().numpy()
     
 def main():
     #torch.cuda.init()
@@ -80,7 +80,7 @@ def main():
     func = functools.partial(image_create, device=device)
     func = functools.update_wrapper(func, image_create)
     iface = gr.Interface(
-          image_create, 
+          func, 
           [
                 gr.inputs.Number(default=0, label='Seed'),
                 gr.inputs.Slider(