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
|