Update app.py

app.py

@@ -17,16 +17,6 @@ def chatbot(prompt):
     # Generate the image based on the user's input
     image = generate_image(prompt, unet)
     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(