Update app.py

app.py

@@ -10,20 +10,22 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
 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)
+  # ... (existing code for encoding the email)
 
-  # Convert tensor to list before concatenation
-  inputs["input_ids"] = [tokenizer.cls_token_id] + inputs["input_ids"].tolist()  
+  # Check length instead of shape
+  if len(inputs["input_ids"]) > 512:  # Adjust maximum sequence length as needed
+    print("WARNING: Input sequence exceeds maximum length. Truncating.")
+    inputs["input_ids"] = inputs["input_ids"][:512]
 
   # Generate questions using model
   generation = model.generate(
       **inputs,  # Unpack the entire inputs dictionary
       max_length=256,  # Adjust max length as needed
-      num_beams=5,  # Adjust beam search for better quality (slower)
-      early_stopping=True,
+      # ... (other generation parameters)
   )
 
+  # ... (existing code for decoding the generation)
+
   # Decode the generated text
   return tokenizer.decode(generation[0], skip_special_tokens=True)