Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import warnings
 import numpy as np
 import pandas as pd
@@ -197,7 +196,11 @@ def predict_emotion(context):
 def generate_response(context):
     tokenizer, model = get_finetuned_lm_model()
     inputs = tokenizer.encode(context, return_tensors='pt')
-    outputs = model.generate(inputs, max_length=500, num_return_sequences=1, pad_token_id=tokenizer.eos_token)
+
+    # Ensure pad_token_id is a tensor
+    pad_token_id = torch.tensor(tokenizer.pad_token_id)
+
+    outputs = model.generate(inputs, max_length=500, num_return_sequences=1, pad_token_id=pad_token_id.item())
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return response