Update main.py

main.py

@@ -8,20 +8,30 @@ 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)
+client = Client("Makhinur/Image_Face_Upscale_Restoration-GFPGAN", hf_token=HF_TOKEN)
+
+
 
 
 @app.post("/upload/")
-async def upload_image(file: UploadFile = File(...)):
-    try:
-        temporary file
+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"
         )
 
@@ -38,3 +48,4 @@ async def upload_image(file: UploadFile = File(...)):
         # Log the error message for debugging
         print(f"Error processing image: {e}")
         return JSONResponse(status_code=500, content={"message": "Internal Server Error"})
+