Skip to content

Commit

Permalink
Update Dockerfile
Browse files Browse the repository at this point in the history
  • Loading branch information
furkmak authored Oct 3, 2024
1 parent c55eae0 commit 150a27e
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -14,10 +14,14 @@ RUN python -c "from transformers import AutoModelForQuestionAnswering, AutoToken
AutoTokenizer.from_pretrained('deepset/roberta-base-squad2', cache_dir='/app/models'); \
AutoModelForQuestionAnswering.from_pretrained('deepset/roberta-base-squad2', cache_dir='/app/models')"

# Copy the fine-tuned model into the container
COPY fine_tuned_legalqa ./fine_tuned_legalqa

# Copy the rest of the application code
COPY . .

# Expose the port that Uvicorn runs on
EXPOSE 8000

CMD ["uvicorn", "fastapp:app", "--host", "0.0.0.0", "--port", "8000"]
# Run the application with Uvicorn
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]

0 comments on commit 150a27e

Please sign in to comment.