dinhquangson commited on
Commit
a7a8d61
·
verified ·
1 Parent(s): 0ef1c26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -11,7 +11,6 @@ from qdrant_client import models, QdrantClient
11
  from itertools import islice
12
 
13
  FILEPATH_PATTERN = "structured_data_doc.parquet"
14
- CACHE_DIR = "cache"
15
  NUM_PROC = os.cpu_count()
16
 
17
 
@@ -71,7 +70,6 @@ async def create_upload_file(file: UploadFile = File(...)):
71
  data_files=FILEPATH_PATTERN,
72
  split="train",
73
  keep_in_memory=True,
74
- cache_dir=CACHE_DIR,
75
  num_proc=NUM_PROC*2)
76
  # Generate and append embeddings to the train split
77
  law_embeddings = generate_embeddings(full_dataset)
 
11
  from itertools import islice
12
 
13
  FILEPATH_PATTERN = "structured_data_doc.parquet"
 
14
  NUM_PROC = os.cpu_count()
15
 
16
 
 
70
  data_files=FILEPATH_PATTERN,
71
  split="train",
72
  keep_in_memory=True,
 
73
  num_proc=NUM_PROC*2)
74
  # Generate and append embeddings to the train split
75
  law_embeddings = generate_embeddings(full_dataset)