File size: 1,390 Bytes
b2bc1d4 602f5d7 b2bc1d4 602f5d7 4bb7c9d 602f5d7 4d3127a 602f5d7 1cf3818 602f5d7 3a9b3a9 1cf3818 602f5d7 1cf3818 b2bc1d4 1cf3818 4f2123c b2bc1d4 1cf3818 602f5d7 b2bc1d4 602f5d7 b2bc1d4 602f5d7 b2bc1d4 602f5d7 b2bc1d4 1cf3818 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
from fastapi import FastAPI, File, UploadFile, Form
from fastapi.responses import JSONResponse
from gradio_client import Client, handle_file
import shutil
import base64
import os
app = FastAPI()
HF_TOKEN = os.getenv("HF_TOKEN")
# Initialize the Gradio client with the token
client = Client("Makhinur/Bringingoldphotoliveagain", hf_token=HF_TOKEN)
@app.post("/upload/")
async def upload_image(
file: UploadFile = File(...)
):
# Save the uploaded image to a temporary file
temp_file_path = "temp_image.png"
with open(temp_file_path, "wb") as buffer:
shutil.copyfileobj(file.file, buffer)
try:
# Use the Gradio client to process the image
result = client.predict(
img=handle_file(temp_file_path),
api_name="/predict"
)
# Read the result image and encode it in base64
with open(result[0], "rb") as img_file:
b64_string = base64.b64encode(img_file.read()).decode('utf-8')
# Clean up the temporary file
os.remove(temp_file_path)
return JSONResponse(content={"sketch_image_base64": f"data:image/png;base64,{b64_string}"})
except Exception as e:
# Log the error message for debugging
print(f"Error processing image: {e}")
return JSONResponse(status_code=500, content={"message": "Internal Server Error"})
|