FilipL009 commited on
Commit
8e698f2
·
verified ·
1 Parent(s): ca64e03

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +3 -2
src/streamlit_app.py CHANGED
@@ -158,8 +158,9 @@ with col2:
158
  progress_bar.progress(20, text="Běží AI model (Batch Processing)...")
159
 
160
  # Zde voláme novou funkci s batch_size
161
- raw_results = analyze_long_text_batched(nlp_pipeline, text_to_analyze, batch_size=8)
162
-
 
163
  # 2. Slepování entit
164
  progress_bar.progress(80, text="Čištění výsledků...")
165
  results = merge_close_entities(raw_results, text_to_analyze)
 
158
  progress_bar.progress(20, text="Běží AI model (Batch Processing)...")
159
 
160
  # Zde voláme novou funkci s batch_size
161
+ # raw_results = analyze_long_text_batched(nlp_pipeline, text_to_analyze, batch_size=8)
162
+ raw_results = analyze_long_text_batched(nlp_pipeline, text_to_analyze, batch_size=1)
163
+
164
  # 2. Slepování entit
165
  progress_bar.progress(80, text="Čištění výsledků...")
166
  results = merge_close_entities(raw_results, text_to_analyze)