Ashraf-CK commited on
Commit
42accf0
·
verified ·
1 Parent(s): 71a8c88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -15,6 +15,8 @@ def root():
15
  return {"message": "Welcome to the medicine Search API!"}
16
  # Load Dataset
17
  dataset = load_dataset("MohamedAshraf701/medicine-dataset", split="train")
 
 
18
 
19
  # Define fields for embedding
20
  fields_for_embedding = [
 
15
  return {"message": "Welcome to the medicine Search API!"}
16
  # Load Dataset
17
  dataset = load_dataset("MohamedAshraf701/medicine-dataset", split="train")
18
+ # Limit the dataset to 30,000 entries
19
+ dataset = dataset.select(range(min(30000, len(dataset))))
20
 
21
  # Define fields for embedding
22
  fields_for_embedding = [