Update app.py

app.py

@@ -19,7 +19,7 @@ def get_token_likelyhoods(model_name="Salesforce/codegen-350M-mono", text="World
     probs = F.softmax(out.logits, dim=-1).squeeze()
     
     output = []
-    for tok, logits in zip(input_ids.squeeze()[1:], probs):
+    for tok, logits in zip(input_ids.squeeze(), probs):
         output.append((
             tokenizer.decode(tok),
             str(round(logits[tok].item() * 100, 4)) + "%",