| def few_shot_pe_llm_0(): | |
| pipe = pipeline("text-classification", model="kolkata97/autotrain-pe-llm-0") | |
| predicted_categories = [] | |
| for sentence in sentences_list: | |
| results = pipe(sentence) | |
| predicted_categories.append(results[0]['label']) | |
| with open(output_csv_file, 'r', newline='', encoding='utf-8') as file: | |
| csv_reader = csv.reader(file) | |
| rows = list(csv_reader) | |
| for i, row in enumerate(rows[1:], start=0): | |
| row.append(predicted_categories[i]) | |
| with open(output_csv_file, 'w', newline='', encoding='utf-8') as file: | |
| writer = csv.writer(file) | |
| writer.writerows(rows) | |
| print("Predicted categories appended to the CSV file.") |