AurelioAguirre commited on
Commit
7055ca8
·
1 Parent(s): a4e24d4

Fixing dockerfile v4

Browse files
Files changed (1) hide show
  1. 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(api,
 
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