Update app.py

app.py

@@ -145,8 +145,9 @@ def process_reviews(df, review_column, class_names):
 
         raw_scores = classify_reviews(df[review_column].tolist())
         for i in range(0, len(raw_scores), BATCH_SIZE):
-            review_counter += BATCH_SIZE
-            progress_bar.progress(review_counter / total_reviews)
+            review_counter += min(BATCH_SIZE, len(raw_scores) - i)  # Avoids overshooting the total reviews count
+            progress = min(review_counter / total_reviews, 1)  # Ensures progress does not exceed 1
+            progress_bar.progress(progress)
 
     with st.spinner('Generating classes...'):
         class_scores = classify_with_new_classes(df[review_column].tolist(), class_names)