Spaces:
Runtime error
Runtime error
Commit
·
7055ca8
1
Parent(s):
a4e24d4
Fixing dockerfile v4
Browse files- src/main.py +3 -1
src/main.py
CHANGED
@@ -33,8 +33,10 @@ def main():
|
|
33 |
# Initialize the router with our config
|
34 |
init_router(config)
|
35 |
api = InferenceApi()
|
|
|
36 |
# Create LitServer instance
|
37 |
-
server = ls.LitServer(
|
|
|
38 |
timeout=config.get("server", {}).get("timeout", 60),
|
39 |
max_batch_size=1,
|
40 |
track_requests=True
|
|
|
33 |
# Initialize the router with our config
|
34 |
init_router(config)
|
35 |
api = InferenceApi()
|
36 |
+
|
37 |
# Create LitServer instance
|
38 |
+
server = ls.LitServer(
|
39 |
+
api,
|
40 |
timeout=config.get("server", {}).get("timeout", 60),
|
41 |
max_batch_size=1,
|
42 |
track_requests=True
|