More fixes.

solution/Dockerfile

@@ -2,11 +2,6 @@
 
 FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime
 
-RUN apt-get update --yes && \
-    apt-get install --yes --no-install-recommends \
-    git \
-    git-lfs
-
 WORKDIR /code
 COPY ./requirements.txt /code/requirements.txt
 RUN pip install -r /code/requirements.txt 

solution/app.py

@@ -4,7 +4,6 @@
 from transformers import pipeline
 from starlette.requests import Request
 
-
 app = FastAPI()
 
 models = {
@@ -16,7 +15,7 @@
 }
 
 for model_key, model_dict in models.items():
-    text_classification_pipeline = pipeline('text-classification', model=model_dict["name"], device=0)
+    text_classification_pipeline = pipeline('text-classification', model=model_dict["name"], device="cuda:0")
     model_dict["pipeline"] = text_classification_pipeline