madebyaris commited on
Commit
391b34c
·
1 Parent(s): 204221f
Files changed (2) hide show
  1. Dockerfile +1 -2
  2. app.py +1 -1
Dockerfile CHANGED
@@ -11,5 +11,4 @@ COPY --chown=user ./ $HOME/app
11
 
12
  RUN pip install -r requirements.txt
13
 
14
-
15
- CMD fastapi run --reload --host=0.0.0.0 --port=7860
 
11
 
12
  RUN pip install -r requirements.txt
13
 
14
+ CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
app.py CHANGED
@@ -3,7 +3,7 @@ from fastapi.responses import HTMLResponse
3
  from fastapi.templating import Jinja2Templates
4
  from transformers import pipeline
5
 
6
- pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1")
7
 
8
  app = FastAPI()
9
  templates = Jinja2Templates(directory="templates")
 
3
  from fastapi.templating import Jinja2Templates
4
  from transformers import pipeline
5
 
6
+ pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1", trust_remote_code=True)
7
 
8
  app = FastAPI()
9
  templates = Jinja2Templates(directory="templates")