Upload model_utils.py

model/model_utils.py

@@ -11,8 +11,8 @@ def load_model():
 
 def generate_explanation(code, tokenizer, model):
     device = model.device
-    # Better prompt engineering
-    input_text = f"summarize: This Python function does the following: {code}"
+    # Final prompt style: generate docstring
+    input_text = f"generate docstring: {code.strip()}"
     input_ids = tokenizer.encode(input_text, return_tensors="pt", truncation=True).to(device)
-    output = model.generate(input_ids, max_new_tokens=200, early_stopping=True)
+    output = model.generate(input_ids, max_new_tokens=150, early_stopping=True)
     return tokenizer.decode(output[0], skip_special_tokens=True)