Upload model_utils.py

model/model_utils.py

@@ -13,5 +13,10 @@ def load_model():
 
 def generate_explanation(prompt, tokenizer, model, device):
     inputs = tokenizer(prompt, return_tensors="pt", truncation=True).to(device)
-    output = model.generate(**inputs, max_new_tokens=256, temperature=0.7)
+    output = model.generate(
+        **inputs,
+        decoder_start_token_id=tokenizer.pad_token_id,
+        max_new_tokens=256,
+        temperature=0.7
+    )
     return tokenizer.decode(output[0], skip_special_tokens=True)