alessandro trinca tornidor commited on
Commit
72ceb76
·
1 Parent(s): d5014a3

[fix] rename app.py to main.py, use empty sys.argv to keep only defaults args

Browse files
Files changed (2) hide show
  1. app.py → main.py +2 -1
  2. scripts/entrypoint.sh +5 -1
app.py → main.py RENAMED
@@ -134,7 +134,8 @@ def get_gradio_interface(fn_inference: Callable):
134
  )
135
 
136
 
137
- args = parse_args(sys.argv[1:])
 
138
  inference_fn = get_inference_model_by_args(args)
139
  io = get_gradio_interface(inference_fn)
140
  app = gr.mount_gradio_app(app, io, path=CUSTOM_GRADIO_PATH)
 
134
  )
135
 
136
 
137
+ logging.info(f"sys.argv:{sys.argv}.")
138
+ args = parse_args([])
139
  inference_fn = get_inference_model_by_args(args)
140
  io = get_gradio_interface(inference_fn)
141
  app = gr.mount_gradio_app(app, io, path=CUSTOM_GRADIO_PATH)
scripts/entrypoint.sh CHANGED
@@ -36,6 +36,9 @@ which nvidia-smi
36
  nvidia-smi
37
  pip list
38
 
 
 
 
39
  df -h / /data /home /var/task
40
 
41
  echo "WORKDIR - /var/task"
@@ -48,6 +51,7 @@ CUDA_VISIBLE_DEVICES=$(nvidia-smi --query-gpu=memory.free,index --format=csv,nou
48
  echo "calculated CUDA_VISIBLE_DEVICES env variable: ${CUDA_VISIBLE_DEVICES}."
49
  export CUDA_VISIBLE_DEVICES
50
 
51
- uvicorn app:app --host 0.0.0.0 --port 7860
 
52
 
53
  exit 0
 
36
  nvidia-smi
37
  pip list
38
 
39
+ which uvicorn
40
+ ls -l ${WORKDIR}/venv/bin/uvicorn
41
+
42
  df -h / /data /home /var/task
43
 
44
  echo "WORKDIR - /var/task"
 
51
  echo "calculated CUDA_VISIBLE_DEVICES env variable: ${CUDA_VISIBLE_DEVICES}."
52
  export CUDA_VISIBLE_DEVICES
53
 
54
+ echo "running command 'uvicorn main:app --host 0.0.0.0 --port 7860'..."
55
+ uvicorn main:app --host 0.0.0.0 --port 7860
56
 
57
  exit 0