Update Dockerfile

Dockerfile

@@ -1,16 +1,13 @@
-
 FROM huggingface/transformers-pytorch-gpu:latest
 
-
 RUN pip install --upgrade pip
-RUN pip install transformers
-
+RUN pip install transformers torch fastapi uvicorn
 
-ENV MODEL_NAME="Hadeel11/your-finetuned-model"
-ENV USE_FP16=True
+COPY app.py /app/app.py
 
+WORKDIR /app
 
 EXPOSE 8080
 
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8080"]
 
-CMD ["python3", "-m", "transformers_serve", "--model-id", "${MODEL_NAME}"]