Tonyivan commited on
Commit
129dbc9
1 Parent(s): 9b74ec6

Update dockerfile

Browse files
Files changed (1) hide show
  1. dockerfile +11 -17
dockerfile CHANGED
@@ -1,22 +1,16 @@
1
- from sentence_transformers import SentenceTransformer
2
- from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
3
 
 
4
 
5
- model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
6
- question_model = "deepset/tinyroberta-squad2"
 
7
 
8
- nlp = pipeline('question-answering', model=question_model, tokenizer=question_model)
9
 
 
 
10
 
11
- def modifyQuery(query_string):
12
- binary_embeddings = model.encode([query_string], precision="binary")
13
- return binary_embeddings[0].tolist()
14
-
15
-
16
- def answerQuestion(question, context):
17
- QA_input = {
18
- 'question': question,
19
- 'context': context
20
- }
21
- res = nlp(QA_input)
22
- return res['answer']
 
1
+ # Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
2
+ # you will also find guides on how best to write your Dockerfile
3
 
4
+ FROM python:3.9
5
 
6
+ RUN useradd -m -u 1000 user
7
+ USER user
8
+ ENV PATH="/home/user/.local/bin:$PATH"
9
 
10
+ WORKDIR /app
11
 
12
+ COPY --chown=user ./requirements.txt requirements.txt
13
+ RUN pip install --no-cache-dir --upgrade -r requirements.txt
14
 
15
+ COPY --chown=user . /app
16
+ CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]