Update app.py

app.py

@@ -12,7 +12,9 @@ def generate_questions(email):
   """Generates questions based on the input email."""
   # Encode the email and prompt together with tokenizer
   inputs = tokenizer(email, return_tensors="pt", add_special_tokens=True)
-  inputs["input_ids"] = [tokenizer.cls_token_id] + inputs["input_ids"]  # Add CLS token at the beginning
+
+  # Convert tensor to list before concatenation
+  inputs["input_ids"] = [tokenizer.cls_token_id] + inputs["input_ids"].tolist()  
 
   # Generate questions using model
   generation = model.generate(
@@ -26,6 +28,7 @@ def generate_questions(email):
   return tokenizer.decode(generation[0], skip_special_tokens=True)
 
 
+
 def generate_answers(questions):
   """Generates possible answers to the input questions."""
   # Encode each question with tokenizer, separated by newline