Update app.py

app.py

@@ -12,12 +12,18 @@ def generate_image(prompt):
 def chatbot(prompt):
     # Generate the image based on the user's input
     image = generate_image(prompt)
-
-
-
-
     return image
 
+def get_aug_embed(self, text_embeds, image):
+    if text_embeds is None:
+        text_embeds = self.text_encoder(
+            text_embeds=text_embeds,
+            image=image,
+            height=self.unet.config.sample_size,
+            width=self.unet.config.sample_size,
+        )
+    return text_embeds
+    
 # Create the Gradio interface
 interface = gr.Interface(
     fn=chatbot,