fix

app.py

@@ -41,22 +41,13 @@ 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, 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"]
+    summary = summarizer(article_text, max_length=100, min_length=30, 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,10 +1,8 @@
 gradio==3.50.2
 transformers==4.42.4
-torch==2.0.1+cu118  # hoặc cu117, tuỳ GPU driver
-torchvision==0.15.2+cu118
-torchaudio==2.0.2+cu118
+torch==2.0.1
 diffusers==0.29.2
 pillow==9.5.0
 safetensors>=0.4.1
 huggingface_hub==0.23.4
-peft==0.12.0
+peft==0.12.0