Update app.py

app.py

@@ -31,7 +31,7 @@ def generate_image(prompt):
 
     generator = torch.Generator(device=device).manual_seed(1024)
     with autocast("cuda"):
-        prompt_embedding = text_embedding_model.encode([str(prompt.item())])[0]
+        prompt_embedding = text_embedding_model.encode([str(prompt])[0]
         prompt_tensor = torch.tensor(prompt_embedding, device=device).half()
         image = pipe(prompt=prompt_tensor, init_image=init_img, strength=0.75, guidance_scale=7.5, generator=generator).images[0]