Update app.py

app.py

@@ -13,7 +13,8 @@ def generate_image(prompt, unet, pipe):
     tokens = pipe.tokenizer(prompt, padding=True, truncation=True, max_length=50, return_tensors="pt").to(unet.device)
     
     # Generate the image
-    image = unet(text_embeddings=tokens.input_ids).images[0]
+    text_embeds = pipe.text_encoder(tokens.input_ids)
+    image = unet(text_embeds=text_embeds).images[0]
     return image
 
 def chatbot(prompt):