Refactor progress bar life cycle

1 files changed, 3 insertions, 6 deletions

diff --git a/llm/perplexity.go b/llm/perplexity.go
index ab31889..5d9056e 100644
--- a/llm/perplexity.go
+++ b/llm/perplexity.go
@@ -1,7 +1,6 @@
 package llm
 
 import (
-	"context"
 	"fmt"
 	"sync"
 	"time"
@@ -47,16 +46,14 @@ func (e *Evaluator[R]) Run(data dataset.Reader, window, stride int) error {
 		}()
 	}
 
-	ctx, cancel := context.WithCancel(context.Background())
-
-	defer cancel()
-
 	pb := tui.NewProgressBar("Perplexity", 20, 0, time.Now())
 
-	go pb.Watch(ctx, 1*time.Second, func() int {
+	pb.Start(1*time.Second, func() int {
 		return int(e.completed.Load())
 	})
 
+	defer pb.Close()
+
 	n := 0
 
 	for d := range data.Texts("text") {