Update app.py

app.py

@@ -77,6 +77,7 @@ def sample_once(prompt, temperature=1.0, top_k=40, max_new=40):
             break
         seq.append(idx)
     generated = [itos[i] for i in seq if i not in {BOS, SEP, EOS, PAD}]
+    print(f"Generated token IDs: {seq}")
     name = ''.join(generated).replace(prompt, "").strip()
     print(f"Sample took: {time.time() - sample_i:.2f}s")
     return clean_name(name)