fixing pydantic v1

main/main.py

@@ -7,7 +7,7 @@ import logging
 from pathlib import Path
 from fastapi.middleware.cors import CORSMiddleware
 from .routes import router, init_router
-from .api import InferenceApi
+from api import InferenceApi
 
 def setup_logging():
     """Set up basic logging configuration"""
@@ -31,9 +31,9 @@ def create_app():
         # Load configuration
         config = load_config()
 
-        # Initialize the router with our config
+        # Initialize API and router
+        api = InferenceApi()
         init_router(config)
-        api = InferenceApi(config)
 
         # Create LitServer instance
         server = ls.LitServer(
@@ -46,10 +46,10 @@ def create_app():
         # Add CORS middleware
         server.app.add_middleware(
             CORSMiddleware,
-            allow_origins=["*"],  # Allows all origins
+            allow_origins=["*"],
             allow_credentials=True,
-            allow_methods=["*"],  # Allows all methods
-            allow_headers=["*"],  # Allows all headers
+            allow_methods=["*"],
+            allow_headers=["*"],
         )
 
         # Add our routes to the server's FastAPI app
@@ -62,12 +62,4 @@ def create_app():
         raise
 
 # Create the application instance
-app = create_app()
-
-if __name__ == "__main__":
-    # If we run this directly with python, we can still use the old method
-    config = load_config()
-    port = config.get("server", {}).get("port", 8001)
-
-    # Get the server instance from our app and run it
-    app.parent.run(port=port)  # assuming the LitServer instance is stored as parent
+app = create_app()