diff --git a/server.py b/server.py
index b899b81..be2e6b1 100644
--- a/server.py
+++ b/server.py
@@ -17,8 +17,7 @@
 
 # io = gr.Interface(lambda x: "Hello, " + x + "!", "textbox", "textbox")
 
-def add(app: FastAPI):
-    app.add_middleware(
+app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
     allow_credentials=True,
@@ -59,4 +58,4 @@ def main():
     uvicorn.run("server:app", host=args.host, port=args.port , reload=args.reload)
 
 if __name__ == "__main__":
-    main()
\ No newline at end of file
+    main()