Update app.py

app.py

@@ -28,6 +28,7 @@ def generate_image(prompt):
 
     generator = torch.Generator(device=device).manual_seed(1024)
     with autocast("cuda"):
+        prompt = torch.tensor(prompt, device=device).half()
         image = pipe(prompt=prompt, init_image=init_img, strength=0.75, guidance_scale=7.5, generator=generator).images[0]
 
     return image