donnyb commited on
Commit
18ea353
·
1 Parent(s): c0b7848

switched to fire

Browse files
Files changed (3) hide show
  1. Dockerfile +1 -1
  2. main.py +2 -3
  3. requirements.txt +3 -2
Dockerfile CHANGED
@@ -24,4 +24,4 @@ WORKDIR $HOME/app
24
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
25
  COPY --chown=user . $HOME/app
26
 
27
- CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
 
24
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
25
  COPY --chown=user . $HOME/app
26
 
27
+ CMD ["python3", "main.py", "--host", "0.0.0.0", "--port", "7860"]
main.py CHANGED
@@ -10,7 +10,7 @@ import pyarrow as pa
10
  from uvicorn import run
11
  from fastapi.staticfiles import StaticFiles
12
 
13
- # from fire import Fire
14
 
15
  app = FastAPI()
16
 
@@ -52,5 +52,4 @@ def serve(port=8000, host="localhost"):
52
 
53
 
54
  if __name__ == "__main__":
55
- # Fire(serve) # so I can run cli args with it
56
- pass
 
10
  from uvicorn import run
11
  from fastapi.staticfiles import StaticFiles
12
 
13
+ from fire import Fire
14
 
15
  app = FastAPI()
16
 
 
52
 
53
 
54
  if __name__ == "__main__":
55
+ Fire(serve) # so I can run cli args with it
 
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  fastapi
2
- uvicorn
3
  duckdb
4
- pyarrow
 
 
1
  fastapi
2
+ uvicorn[standard]
3
  duckdb
4
+ pyarrow
5
+ fire