Update app.py

app.py

@@ -18,11 +18,11 @@ translation_models = {
 # Initialize summarization pipelines with specified models
 summarization_models = {
     'Scientific': "facebook/bart-large-cnn",
-    'Literature': "t5-small"
+    'Literature': "allenai/scibert_scivocab_cased"
 }
 
 # Initialize tokenizer
-tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")  # Using BART tokenizer for the 'Scientific' model
+tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
 
 # Initialize summarization pipelines
 summarizers = {model: pipeline("summarization", model=model) for model in summarization_models.values()}
@@ -60,7 +60,7 @@ def summarize_text(text, model):
     summaries = []
     for chunk in chunks:
         try:
-            summary = summarizers[model](chunk, max_length=150, min_length=40, do_sample=False)[0]['summary_text']
+            summary = summarizers[model](chunk, max_length=150, min_length=20, do_sample=False)[0]['summary_text']
             summaries.append(summary)
         except Exception as e:
             print(f"Error summarizing chunk: {chunk}\nError: {e}")
@@ -142,3 +142,4 @@ iface.launch()
 
 
 
+