changing port to 7860 and adding gpu support

Dockerfile

@@ -8,4 +8,4 @@ RUN pip install -r requirements.txt
 
 COPY . .
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -7,8 +7,9 @@ from enum import Enum
 from transformers import M2M100Tokenizer, M2M100ForConditionalGeneration
 import torch
 
-app = FastAPI()
-device = torch.device("cpu")
+app = FastAPI(docs_url="/")
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+
 
 
 class TranslationRequest(BaseModel):
@@ -47,4 +48,6 @@ async def translate(request: TranslationRequest):
 if __name__ == "__main__":
     import uvicorn
 
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    uvicorn.run(app, host="0.0.0.0", port=7860)
+
+