Update app.py

app.py

@@ -46,6 +46,7 @@ elif inpt and file is None and not("Generate" in inpt or "Draw" in inpt or "Imag
     aimsg.write(output)
 
 elif inpt and ("Generate" in inpt or "Draw" in inpt or "Imagine" in inpt):
+    prompt = inpt.split("--")[1]
     payload = {"prompts": prompt}
     output = generate_image(payload)
     if output: