Renamed variables

app.py

@@ -24,9 +24,9 @@ model = GPTLanguageModel(vocab_size)
 model.load_state_dict(torch.load('gpt_model_saved.pth', map_location=cfg.device))
 m = model.to(cfg.device)
 
-def inference(input_context, count):
-    encoded_text = [encode(input_context)]
-    count = int(count)
+def inference(InputContext, DesiredCharacterCount):
+    encoded_text = [encode(InputContext)]
+    count = int(DesiredCharacterCount)
     context = torch.tensor(encoded_text, dtype=torch.long, device=cfg.device)
      
     out_text = decode(m.generate(context, max_new_tokens=count)[0].tolist())