Update app.py

app.py

@@ -11,15 +11,15 @@ data_collator = DataCollatorForLanguageModeling(tokenizer=__tokenizer, mlm=False
 def queryGPT(question):
     return generate_response(__model, __tokenizer, question)
 
-def generate_response(__model, __tokenizer, prompt, max_length=200):
+def generate_response(model,tokenizer, prompt, max_length=200):
 
     input_ids = tokenizer.encode(prompt, return_tensors="pt")      # 'pt' for returning pytorch tensor
 
     # Create the attention mask and pad token id
     attention_mask = torch.ones_like(input_ids)
-    pad_token_id = __tokenizer.eos_token_id
+    pad_token_id = tokenizer.eos_token_id
 
-    output = __model.generate(
+    output = model.generate(
         input_ids,
         max_length=max_length,
         num_return_sequences=1,
@@ -27,7 +27,7 @@ def generate_response(__model, __tokenizer, prompt, max_length=200):
         pad_token_id=pad_token_id
     )
 
-    return __tokenizer.decode(output[0], skip_special_tokens=True)
+    return tokenizer.decode(output[0], skip_special_tokens=True)
     
 with gr.Blocks() as demo: