jiminaa commited on
Commit
b2e7508
·
1 Parent(s): 73eb2b3
Files changed (2) hide show
  1. Dockerfile +1 -0
  2. requirements.txt +1 -1
Dockerfile CHANGED
@@ -10,5 +10,6 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
10
 
11
  USER user
12
 
 
13
  # Start the FastAPI app on port 7860, the default port expected by Spaces
14
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
 
10
 
11
  USER user
12
 
13
+
14
  # Start the FastAPI app on port 7860, the default port expected by Spaces
15
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
  --extra-index-url https://download.pytorch.org/whl/cpu
2
  torch
3
  peft
4
- gradio >= 4.36
5
  transformers
6
  accelerate
7
  fastapi
 
1
  --extra-index-url https://download.pytorch.org/whl/cpu
2
  torch
3
  peft
4
+ gradio>=4.36,<5.0
5
  transformers
6
  accelerate
7
  fastapi