google

app.py

@@ -74,13 +74,19 @@ def summarize_text(input_text, model_label, char_limit):
     inputs = tokenizer(input_text, return_tensors="pt", truncation=True)
     input_ids = inputs["input_ids"].to(device)
 
+    # Adjust the length constraints to make sure min_length < max_length
+    max_len = 30  # Set your desired max length
+    min_len = 5   # Ensure min_length is smaller than max_length
+
+    # Generate summary
     summary_ids = model.generate(
         input_ids,
-        max_length=30,  # Ensure max_length is greater than min_length
-        min_length=5,   # Ensure min_length is less than max_length
+        max_length=max_len,
+        min_length=min_len,
         do_sample=False
     )
 
+    # Decode the summary
     summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
     return summary[:char_limit].strip()