Upload 2 files

app.py

@@ -13,7 +13,7 @@ def predict_batch_from_csv(input_file, output_file):
 
     # Iterate over rows with tqdm for progress tracking
     for _, row in tqdm(batch_data.iterrows(), total=len(batch_data)):
-        text = row['pnc_fac_name']  # Replace 'facility_name' with the actual column name containing the text data
+        text = row['facility_name']  # Replace 'facility_name' with the actual column name containing the text data
 
         if pd.isnull(text):
             cleaned_text = ""

facility_predict.py

@@ -164,7 +164,7 @@ def predict_batch_from_csv(input_file, output_file):
     # Iterate over rows with tqdm for progress tracking
     # Iterate over rows with tqdm for progress tracking
     for _, row in tqdm(batch_data.iterrows(), total=len(batch_data)):
-        text = row['pnc_fac_name']  # Replace 'facility_name' with the actual column name containing the text data
+        text = row['facility_name']  # Replace 'facility_name' with the actual column name containing the text data
 
         if pd.isnull(text):
             cleaned_text = ""