Update app.py

app.py

@@ -23,6 +23,7 @@ from typing import Annotated
 from transformers import BertTokenizerFast, EncoderDecoderModel
 import torch
 import random
+from fastapi import FastAPI, Response
 import string
 import time
 from huggingface_hub import InferenceClient
@@ -178,5 +179,25 @@ def do_ML(filename:str,text:str,code:str,host:str):
         data={"text":text,"filename":filename}
         requests.post(host+"texttoimage2handleerror",data=data)
         
+@app.post("/image")
+async def get_answer(q: Query ):
+    text = q.text
+    try:
+        global client
+        imagei = client.text_to_image(text)
+        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})
+    
+    
+
+    
+
+
+