Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
16
  # -------------------
17
  # Config
18
  # -------------------
19
- MODEL_NAME = "your-model-name" # change to your embedding model
20
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
21
  BATCH_SIZE = 16
22
  BATCH_TIMEOUT = 0.01
 
16
  # -------------------
17
  # Config
18
  # -------------------
19
+ MODEL_NAME = "jina-embeddings-v4" # change to your embedding model
20
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
21
  BATCH_SIZE = 16
22
  BATCH_TIMEOUT = 0.01