Update app.py

app.py

@@ -400,7 +400,7 @@ def generate_text_with_probs(initial_context, context_text , top_p, max_length,
     Tokens = {}
 
     #input_ids = tokenizer.encode(initial_context, return_tensors="pt").to(device='mps')
-    input_ids = tokenizer.encode(initial_text, return_tensors="pt").to(device='cpu')
+    input_ids = tokenizer.encode(initial_context, return_tensors="pt").to(device='cpu')
     generated_text = initial_context
     token_tables = []