gleisonnanet commited on
Commit
02d730b
·
1 Parent(s): 4d02b11
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. main.py +0 -1
Dockerfile CHANGED
@@ -18,4 +18,4 @@ VOLUME /code/models
18
  COPY . .
19
  RUN transformers-cli download succinctly/text2image-prompt-generator
20
 
21
- CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]
 
18
  COPY . .
19
  RUN transformers-cli download succinctly/text2image-prompt-generator
20
 
21
+ CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
main.py CHANGED
@@ -29,5 +29,4 @@ async def text(text: str):
29
  return {"error": str(e)}
30
 
31
  if __name__ == "__main__":
32
- import uvicorn
33
  uvicorn.run(app, host="0.0.0.0", port=7860)
 
29
  return {"error": str(e)}
30
 
31
  if __name__ == "__main__":
 
32
  uvicorn.run(app, host="0.0.0.0", port=7860)