Feat : better file system interaction

2025-03-09 12:45:29 +01:00
parent 06d207da53
commit 7710fb3f9d
13 changed files with 204 additions and 50 deletions
@@ -118,6 +118,7 @@ class Memory():
            early_stopping=True  # Stop when all beams finish
        )
        summary = self.tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+        summary.replace('summary:', '')
        return summary
    
    @timer_decorator