Update main.py

main.py

@@ -4,17 +4,19 @@ from gradio_client import Client, handle_file
 import shutil
 import os
 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(...)):
+    if not file:
+        raise HTTPException(status_code=400, detail="No file uploaded")
+
     # Save the uploaded file to a temporary location
     temp_file_path = f"temp_{file.filename}"
     with open(temp_file_path, "wb") as buffer: