Update app.py

app.py

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