Hugging Face's logo

Spaces:
JustKiddo
/
IOTraining
Sleeping

App Files Community
JustKiddo commited on
Commit
79f13c5
•
1 Parent(s): b847330

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ class CompanyKnowledgeBase:
9
  def __init__(self, dataset_name="JustKiddo/IODataset"):
10
  # Load dataset from Hugging Face
11
  try:
12
- self.dataset = load_dataset(dataset_name)['company_knowledge']
13
 
14
  # Initialize semantic search
15
  self.model = SentenceTransformer('paraphrase-multilingual-MiniLM-L12-v2')
 
9
  def __init__(self, dataset_name="JustKiddo/IODataset"):
10
  # Load dataset from Hugging Face
11
  try:
12
+ self.dataset = load_dataset(dataset_name)['train']
13
 
14
  # Initialize semantic search
15
  self.model = SentenceTransformer('paraphrase-multilingual-MiniLM-L12-v2')