Update app.py

app.py

@@ -37,7 +37,10 @@ def rewrite_text(text, style):
         prompt = f"Rewrite the following text:\n\n{text}"
 
     # Call the model (paraphraser)
-    response = paraphraser(prompt, max_length=100)[0]['generated_text']
+    inputs = tokenizer(prompt, return_tensors="pt", padding=True, truncation=True)
+    outputs = model.generate(**inputs, max_length=100, num_beams=5, num_return_sequences=1)
+    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+
 
     # Optional: clean output (remove repeated prompt from response if needed)
     return response.replace(prompt, "").strip()