heymenn commited on
Commit
290a90d
·
verified ·
1 Parent(s): 45c5f49

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -23,7 +23,7 @@ app = FastAPI(
23
 
24
 
25
  model = sentence_transformers.SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
26
- dataset = load_dataset("heymenn/Technologies", split="train")
27
 
28
 
29
  class SearchInput(BaseModel):
 
23
 
24
 
25
  model = sentence_transformers.SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
26
+ dataset = load_dataset("heymenn/Technologies", streaming=True, split="train")
27
 
28
 
29
  class SearchInput(BaseModel):