Fix

app.py

@@ -41,13 +41,22 @@ image_generator = StableDiffusionPipeline.from_pretrained(
 lora_weights = "lacos03/std-1.5-lora-midjourney-1.0"
 image_generator.load_lora_weights(lora_weights)
 
-
+# Cấu hình beam search cho BART
+NUM_BEAMS = 10
+NO_REPEAT_NGRAM_SIZE = 3
+LENGTH_PENALTY = 1.0
+MIN_NEW_TOKENS = 10
+MAX_NEW_TOKENS = 62
 # === Hàm xử lý ===
 def summarize_article(article_text):
     """Tóm tắt bài viết và tạo prompt refinement"""
     if not article_text.strip():
         return "[Empty]", "[Empty]"
-    summary = summarizer(article_text, max_length=100, min_length=30, do_sample=False)[0]["summary_text"]
+    summary = summarizer(article_text, num_beams=NUM_BEAMS,
+        no_repeat_ngram_size=NO_REPEAT_NGRAM_SIZE,
+        length_penalty=LENGTH_PENALTY,
+        min_new_tokens=MIN_NEW_TOKENS,
+        max_new_tokens=MAX_NEW_TOKENS, do_sample=False)[0]["summary_text"]
     title = summary.split(".")[0] + "."
     # Prompt refinement
     input_ids = promptist_tokenizer(title.strip() + " Rephrase:", return_tensors="pt").input_ids.to(device)

requirements.txt

@@ -1,6 +1,6 @@
 gradio==3.50.2
 transformers==4.42.4
-torch==2.0.1
+torch==2.4.1
 diffusers==0.29.2
 pillow==9.5.0
 safetensors>=0.4.1