change model

app.py

@@ -23,7 +23,9 @@ model = GroqModel('llama-3.1-70b-versatile', api_key = api_key)
 
 
 # to summarize
-summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+#summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+summarizer = pipeline('text2text-generation', model='describeai/gemini')
+#nlpaueb/legal-bert-base-uncased
 
 
 def split_long_string(long_string, chunk_size=3500):
@@ -70,9 +72,10 @@ async def ppt_content(data):
 
 def ai_ppt(data):
     #call summerizer to summerize pdf 
-    summary = summarizer(data, max_length=400, min_length=100, truncation=True)
+    summary = summarizer("Summarize this text: "+data, max_length=400, min_length=100, truncation=True,do_sample=False)
     
-    summary_texts = [item['summary_text'] for item in summary]
+    #summary_texts = [item['summary_text'] for item in summary]
+    summary_texts = [item['generated_text'] for item in summary]
     asyncio.run(ppt_content(data=summary_texts))