debug

app.py

@@ -2,7 +2,7 @@ import os
 import pickle as pkl
 from pathlib import Path
 from threading import Thread
-from typing import List, Tuple, Iterator
+from typing import List, Tuple, Iterator, Optional
 from queue import Queue
 
 import spaces
@@ -209,7 +209,7 @@ def generate(
         acc_probe_pred = (1 - acc_probe.predict_proba(acc_concat_layers.reshape(1, -1))[0][1]) * 2 - 1
         
         # decode latest token
-        new_test = tokenizer.decode(token_id)
+        new_text = tokenizer.decode(token_id)
         print(new_text, se_probe_pred, acc_probe_pred)  
 
         se_new_highlighted_text = highlight_text(new_text, se_probe_pred)