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/")
@@ -39,8 +41,11 @@ async def upload_file(file: UploadFile = File(...)):
             }
         else:
             return {"error": "Invalid result from the prediction API."}
+    except Exception as e:
+        return {"error": str(e)}
     finally:
-        os.unlink(temp_file_path)
+        if os.path.exists(temp_file_path):
+            os.unlink(temp_file_path)
 
 
 app.mount("/", StaticFiles(directory="static", html=True), name="static")