Update app.py

app.py

@@ -10,7 +10,7 @@ import io
 import base64
 from stqdm import stqdm
 import nltk
-
+import gc
 from nltk.corpus import stopwords
 nltk.download('stopwords')
 import matplotlib.pyplot as plt
@@ -36,6 +36,9 @@ def classify_reviews(reviews):
         inputs = tokenizer(reviews[i:i+BATCH_SIZE], return_tensors='pt', truncation=True, padding=True, max_length=512)
         outputs = model(**inputs)
         probabilities.extend(F.softmax(outputs.logits, dim=1).tolist())
+        del inputs
+        del outputs
+        gc.collect()  # manually invoke garbage collector here
     return probabilities
 
 def top_rating(scores):