Update main.py

main.py

@@ -16,6 +16,8 @@ client = Client("https://ashrafb-image-to-sketch.hf.space/", hf_token=hf_token)
 
 import tempfile
 
+import base64
+
 @app.post("/upload/")
 async def upload_file(file: UploadFile = File(...)):
     with tempfile.NamedTemporaryFile(delete=False) as temp_file:
@@ -24,10 +26,16 @@ async def upload_file(file: UploadFile = File(...)):
 
     try:
         result = client.predict(temp_file_path, api_name="/predict")
-        return {"sketch_image": result[0]}
+        
+        # Convert the image data to base64 string
+        with open(result[0], "rb") as image_file:
+            image_data = base64.b64encode(image_file.read()).decode("utf-8")
+
+        return {"sketch_image_base64": image_data}
     finally:
         os.unlink(temp_file_path)
 
+
 app.mount("/", StaticFiles(directory="static", html=True), name="static")
 
 @app.get("/")