Wider prediction column; dtype kwarg

app.py

@@ -90,7 +90,7 @@ class Engine:
         if self.tok.pad_token is None:
             self.tok.pad_token = self.tok.eos_token
         self.bos = self.tok.bos_token_id if self.tok.bos_token_id is not None else self.tok.eos_token_id
-        self.model = AutoModelForCausalLM.from_pretrained(BASE_MODEL, torch_dtype=DTYPE).to(DEVICE).eval()
+        self.model = AutoModelForCausalLM.from_pretrained(BASE_MODEL, dtype=DTYPE).to(DEVICE).eval()
 
         path = hf_hub_download(CKPT_REPO, CKPT_FILE)
         try:
@@ -455,8 +455,8 @@ CSS = """
 .bars .bar i {height: 8px; border-radius: 4px; background: linear-gradient(90deg, #6d7cff, #22d3b6)}
 table.preds {width: 100%; border-collapse: collapse; margin-top: 6px}
 table.preds td {padding: 8px 10px; border-bottom: 1px solid #eef0f5; vertical-align: top}
-table.preds td.who {width: 170px; font-weight: 600; white-space: nowrap}
-table.preds td.note {color: #8a91a5; font-size: 0.82rem}
+table.preds td.who {width: 150px; font-weight: 600; white-space: nowrap}
+table.preds td.note {width: 130px; color: #8a91a5; font-size: 0.8rem}
 table.preds tr.truth {background: #f7f8fb}
 table.preds tr.c2l {background: #f4f5ff}
 table.preds tr.prev {background: #fffaf2}

requirements.txt

@@ -1,6 +1,6 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
 torch>=2.3
-transformers>=4.45,<5
+transformers>=4.56,<5
 accelerate>=0.34
 huggingface_hub>=0.28
 numpy>=1.24