Update app.py

app.py

@@ -38,7 +38,7 @@ if args.enable_model_cpu_offload:
 else:
     pipe = pipe.to("cuda")
 
-@spaces.GPU
+# @spaces.GPU
 def infer(edit_images, 
           prompt, 
           seed=666, 
@@ -82,7 +82,7 @@ def infer(edit_images,
         width=width*2,
         guidance_scale=guidance_scale,
         num_inference_steps=num_inference_steps,
-        generator=torch.Generator("cpu").manual_seed(seed)
+        generator=torch.Generator().manual_seed(seed),
     ).images[0]
 
     w,h = image.size