mpnikhil commited on
Commit
1f5fac6
·
verified ·
1 Parent(s): 777945f

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. Dockerfile +3 -3
  2. server/app.py +2 -2
Dockerfile CHANGED
@@ -72,10 +72,10 @@ ENV PATH="/app/.venv/bin:$PATH"
72
  ENV PYTHONPATH="/app/env:$PYTHONPATH"
73
 
74
  # Health check
75
- HEALTHCHECK --interval=30s --timeout=5s --start-period=30s --retries=3 \
76
- CMD curl -f http://localhost:7860/health || exit 1
77
 
78
  # Run the FastAPI server
79
  # The module path is constructed to work with the /app/env structure
80
  ENV ENABLE_WEB_INTERFACE=true
81
- CMD ["sh", "-c", "cd /app/env && uvicorn server.app:app --host 0.0.0.0 --port 7860"]
 
72
  ENV PYTHONPATH="/app/env:$PYTHONPATH"
73
 
74
  # Health check
75
+ HEALTHCHECK --interval=30s --timeout=3s --start-period=5s --retries=3 \
76
+ CMD curl -f http://localhost:8000/health || exit 1
77
 
78
  # Run the FastAPI server
79
  # The module path is constructed to work with the /app/env structure
80
  ENV ENABLE_WEB_INTERFACE=true
81
+ CMD ["sh", "-c", "cd /app/env && uvicorn server.app:app --host 0.0.0.0 --port 8000"]
server/app.py CHANGED
@@ -23,7 +23,7 @@ app = create_app(
23
  )
24
 
25
 
26
- def main(host: str = "0.0.0.0", port: int = 7860):
27
  """Entry point for direct execution."""
28
  import uvicorn
29
  uvicorn.run(app, host=host, port=port)
@@ -32,6 +32,6 @@ def main(host: str = "0.0.0.0", port: int = 7860):
32
  if __name__ == "__main__":
33
  import argparse
34
  parser = argparse.ArgumentParser()
35
- parser.add_argument("--port", type=int, default=7860)
36
  args = parser.parse_args()
37
  main(port=args.port)
 
23
  )
24
 
25
 
26
+ def main(host: str = "0.0.0.0", port: int = 8000):
27
  """Entry point for direct execution."""
28
  import uvicorn
29
  uvicorn.run(app, host=host, port=port)
 
32
  if __name__ == "__main__":
33
  import argparse
34
  parser = argparse.ArgumentParser()
35
+ parser.add_argument("--port", type=int, default=8000)
36
  args = parser.parse_args()
37
  main(port=args.port)