Update main.py

from fastapi import FastAPI, File, UploadFile
from fastapi.responses import StreamingResponse
from gradio_client import Client

app = FastAPI()

client = Client("https://ashrafb-image-to-sketch.hf.space/")



@app
.post("/predict")
async def predict_sketch(file: UploadFile = File(...)):
content = await file.read()
# Call the Gradio client to get the sketch result
result = client.predict(content, api_name="/predict")
sketch_image_bytes = result[0].encode('utf-8') # Assuming result[0] is a base64 encoded image
return StreamingResponse(io.BytesIO(sketch_image_bytes), media_type="image/png")

main.py

@@ -3,9 +3,10 @@ from fastapi import FastAPI, File, UploadFile, Form, Request
 from fastapi.responses import HTMLResponse, FileResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
+from fastapi.responses import StreamingResponse
 from gradio_client import Client
 import os
-
+import io
 app = FastAPI()
 
 hf_token = os.environ.get('HF_TOKEN')
@@ -16,7 +17,8 @@ async def predict_sketch(file: UploadFile = File(...)):
     content = await file.read()
     # Call the Gradio client to get the sketch result
     result = client.predict(content, api_name="/predict")
-    return {"sketch_image": result[0], "result_file": result[1]}
+    sketch_image_bytes = result[0].encode('utf-8')  # Assuming result[0] is a base64 encoded image
+    return StreamingResponse(io.BytesIO(sketch_image_bytes), media_type="image/png")
 
 app.mount("/", StaticFiles(directory="static", html=True), name="static")