SKaup commited on
Commit
d792788
·
1 Parent(s): abb67ab

Changed app.py to have print statement

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -0
  2. app.py +1 -0
Dockerfile CHANGED
@@ -8,4 +8,6 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
8
 
9
  COPY . .
10
 
 
 
11
  #CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]
 
8
 
9
  COPY . .
10
 
11
+ CMD ["python3 app.py"]
12
+
13
  #CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]
app.py CHANGED
@@ -1,2 +1,3 @@
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
2
 
 
 
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
2
 
3
+ print(AutoModelForCausalLM)