downgrade python

Dockerfile

@@ -1,5 +1,5 @@
-FROM python:3.9
-
+# FROM python:3.9
+FROM python:3.7-slim-buster
 
 WORKDIR /code
 

main.py

@@ -8,7 +8,7 @@ from pydantic import BaseModel
 from enum import Enum
 from transformers import M2M100Tokenizer, M2M100ForConditionalGeneration, AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
-
+import uvicorn
 app = FastAPI(docs_url="/")
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
@@ -67,7 +67,7 @@ tokenizer = AutoTokenizer.from_pretrained(chat_model_name)
 modelchat = AutoModelForSeq2SeqLM.from_pretrained(chat_model_name)
 
 @app.get("/chat")
-async def read_root(text: str, ):
+async def read_root(text: str):
     input_ids = tokenizer(
         [WHITESPACE_HANDLER(text)],
         return_tensors="pt",
@@ -98,8 +98,6 @@ async def read_root(text: str, ):
 
 
 if __name__ == "__main__":
-    import uvicorn
-
     uvicorn.run(app, host="0.0.0.0", port=7860)