Fix

app.py

@@ -294,7 +294,7 @@ def run_context_length_probing(
                 start=0,
                 pad_id=_tokenizer.eos_token_id
             ).convert_to_tensors("pt")
-            logprobs = get_logprobs(model=model, inputs=inputs_sliding, metric=metric)
+            logprobs = get_logprobs(model=_model, inputs=inputs_sliding, metric=metric)
             output_ids = [*input_ids, label_ids[-1]]
         num_tgt_tokens = logprobs.shape[0]