aibmedia commited on
Commit
a7d861a
·
1 Parent(s): a72117c
Files changed (2) hide show
  1. Dockerfile +2 -2
  2. main.py +2 -1
Dockerfile CHANGED
@@ -10,8 +10,8 @@ COPY ./requirements.txt /code/requirements.txt
10
  # Install requirements.txt
11
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
12
 
13
- RUN pip install ollama --no-cache-dir --upgrade
14
- R
15
 
16
  # Llama copy chmod
17
  #COPY ./mxbai-embed-large-v1-f16.llamafile /code/mxbai-embed-large-v1-f16.llamafile
 
10
  # Install requirements.txt
11
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
12
 
13
+ RUN pip install ollama
14
+
15
 
16
  # Llama copy chmod
17
  #COPY ./mxbai-embed-large-v1-f16.llamafile /code/mxbai-embed-large-v1-f16.llamafile
main.py CHANGED
@@ -17,10 +17,11 @@ app = Flask(__name__)
17
  @app.route('/')
18
  def server_home():
19
  url = 'http://localhost:11434/api/generate'
20
-
21
  form = {'model': 'mxbai-embed-large', 'prompt': 'Give the cosine similarity for each 1.Its a rainy day. 2. The news says it will rain 3. The cat pass across the street'}
22
 
23
  x = requests.post(url, json = form)
 
24
  return str(x)
25
 
26
 
 
17
  @app.route('/')
18
  def server_home():
19
  url = 'http://localhost:11434/api/generate'
20
+
21
  form = {'model': 'mxbai-embed-large', 'prompt': 'Give the cosine similarity for each 1.Its a rainy day. 2. The news says it will rain 3. The cat pass across the street'}
22
 
23
  x = requests.post(url, json = form)
24
+
25
  return str(x)
26
 
27