Changing to smaller model - getting out of memory error

app.py

@@ -9,10 +9,10 @@ os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 
 def summarize(text):
     # Setting to use the bart-large-cnn model for summarization
-    summarizer = pipeline("summarization")
+    # summarizer = pipeline("summarization")
 
     # To use the t5-base model for summarization:
-    # summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base", framework="tf")
+    summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base", framework="tf")
 
     summary_text = summarizer(text, max_length=100, min_length=5, do_sample=False)[0]['summary_text']
     print(f'Length of initial text: {len(text)}')