Update app.py

app.py

@@ -144,7 +144,20 @@ async def get_answer(q: QueryM ):
 
 
 
-
+@app.post("/image_new")
+async def get_answer(q: Query ):
+    text = q.text
+    try:
+        global client
+        imagei = client.post(json={"inputs": text}, model='openskyml/dalle-3-xl')
+        byte_array = io.BytesIO()
+        imagei.save(byte_array, format='JPEG')
+        response = Response(content=byte_array.getvalue(), media_type="image/png")
+        return response
+    
+    except:
+        return JSONResponse({"status":False})
+