Update app.py
#12
by
Amlan99
- opened
app.py
CHANGED
@@ -16,7 +16,7 @@ os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
|
|
16 |
# -------------------
|
17 |
# Config
|
18 |
# -------------------
|
19 |
-
MODEL_NAME = "
|
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
|