Update app.py

app.py

@@ -40,10 +40,12 @@ async def generate_ad(image_file:UploadFile, hex_code:str, prompt: str = Query(.
     
     image_bytes = await image_file.read()
     init_image = Image.open(io.BytesIO(image_bytes)).convert("RGBA")
-    init_image.thumbnail((768, 768))
-    # Generate the image using the text-to-image model
+    #init_image.thumbnail((768, 768))
+    #Generate the image using the text-to-image model
     generator = torch.Generator(device=device1).manual_seed(1024)
-    ad_prompt = f"From init image create me a new image that will attract customers to put in a (ad template) and use this hex code {hex_code} in the image"
+    ad_prompt = f"""Your system prompt is this: {prompt} Consider your system prompt first. 
+                    Then from init image create me a new image that will attract customers to put in a (ad template) 
+                    Also, use this hex code {hex_code} in the image"""
     
     image = pipe(prompt=ad_prompt, image=init_image, strength=0.75, guidance_scale=7.5, generator=generator).images[0]