Update app.py

app.py

@@ -25,7 +25,8 @@ emotion_labels = ["admiration", "amusement", "anger", "annoyance", "approval",
 def classify_emotions_in_batches(texts, batch_size=64):
     results = []
     start_time = time.time()
-    for i in range(0, 1280, batch_size):
+    num_batches = min(20, (len(texts) + batch_size - 1) // batch_size)  # Calculate the number of batches to run (up to 20)
+    for i in range(0, num_batches * batch_size, batch_size):
         batch = texts[i:i+batch_size]
         inputs = tokenizer(batch, return_tensors="pt", truncation=True, padding=True).to(device)
         with torch.no_grad():
@@ -36,7 +37,7 @@ def classify_emotions_in_batches(texts, batch_size=64):
         
         # Log progress
         batch_time = time.time() - start_time
-        st.write(f"Processed batch {i//batch_size + 1} of {len(texts)//batch_size + 1} in {batch_time:.2f} seconds")
+        st.write(f"Processed batch {i//batch_size + 1} of {num_batches} in {batch_time:.2f} seconds")
         start_time = time.time()
     return results