chrissoria commited on
Commit
0905db0
·
verified ·
1 Parent(s): 1e58ec7

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +3 -3
  2. requirements.txt +1 -1
app.py CHANGED
@@ -555,7 +555,7 @@ def run_classify_data(input_type, input_data, description, categories,
555
  if mode:
556
  classify_kwargs['pdf_mode'] = mode
557
 
558
- result = catllm.multi_class_ensemble(**classify_kwargs)
559
 
560
  processing_time = time.time() - start_time
561
  num_items = len(result)
@@ -1058,7 +1058,7 @@ with col_input:
1058
  status_text.text(f"Processing page {current_idx+1} of {total_pages} ({page_label}) ({progress*100:.0f}%){eta_str}")
1059
 
1060
  try:
1061
- result_df = catllm.multi_class_ensemble(
1062
  survey_input=items_list,
1063
  categories=categories_entered,
1064
  models=[(model, model_source, actual_api_key)],
@@ -1113,7 +1113,7 @@ with col_input:
1113
  status_text.text(f"Processing item {current_idx+1} of {total} ({progress*100:.0f}%){eta_str}")
1114
 
1115
  try:
1116
- result_df = catllm.multi_class_ensemble(
1117
  survey_input=items_list,
1118
  categories=categories_entered,
1119
  models=[(model, model_source, actual_api_key)],
 
555
  if mode:
556
  classify_kwargs['pdf_mode'] = mode
557
 
558
+ result = catllm.classify_ensemble(**classify_kwargs)
559
 
560
  processing_time = time.time() - start_time
561
  num_items = len(result)
 
1058
  status_text.text(f"Processing page {current_idx+1} of {total_pages} ({page_label}) ({progress*100:.0f}%){eta_str}")
1059
 
1060
  try:
1061
+ result_df = catllm.classify_ensemble(
1062
  survey_input=items_list,
1063
  categories=categories_entered,
1064
  models=[(model, model_source, actual_api_key)],
 
1113
  status_text.text(f"Processing item {current_idx+1} of {total} ({progress*100:.0f}%){eta_str}")
1114
 
1115
  try:
1116
+ result_df = catllm.classify_ensemble(
1117
  survey_input=items_list,
1118
  categories=categories_entered,
1119
  models=[(model, model_source, actual_api_key)],
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  streamlit>=1.32.0
2
- cat-llm[pdf]>=0.1.11
3
  mistralai
4
  pydantic==2.10.6
5
  huggingface_hub<0.27.0
 
1
  streamlit>=1.32.0
2
+ cat-llm[pdf]>=0.1.12
3
  mistralai
4
  pydantic==2.10.6
5
  huggingface_hub<0.27.0