michaelfeil commited on
Commit
34ece89
1 Parent(s): b958639

push demo again

Browse files
Files changed (3) hide show
  1. Dockerfile +1 -1
  2. entrypoint.sh +5 -0
  3. search.py +0 -4
Dockerfile CHANGED
@@ -5,4 +5,4 @@ RUN pip install groq gradio infinity_emb[all] usearch datasets
5
  WORKDIR /app
6
  COPY . .
7
 
8
- CMD python main.py
 
5
  WORKDIR /app
6
  COPY . .
7
 
8
+ CMD entrypoint.sh
entrypoint.sh ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ export HF_HOME=/tmp/cache
2
+ export TRANSFORMERS_CACHE=/tmp/cache
3
+ export INFINITY_QUEUE_SIZE=512
4
+
5
+ python main.py
search.py CHANGED
@@ -3,10 +3,6 @@ import numpy as np
3
  from usearch.index import Index, Matches
4
  import asyncio
5
  import pandas as pd
6
- import os
7
- os.environ["HF_HOME"] = "/app"
8
- os.environ["TRANSFORMERS_CACHE"] = "/app"
9
- os.environ["INFINITY_QUEUE_SIZE"] = "512000"
10
 
11
  engine = AsyncEmbeddingEngine.from_args(
12
  EngineArgs(
 
3
  from usearch.index import Index, Matches
4
  import asyncio
5
  import pandas as pd
 
 
 
 
6
 
7
  engine = AsyncEmbeddingEngine.from_args(
8
  EngineArgs(