Update app.py

app.py

@@ -27,10 +27,13 @@ def generate_roast(resume_text):
     prompt = f"Roast this resume:\n\n{resume_text}\n\nRoast:"
     
     generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
-    roast = generator(prompt, max_length=150, num_return_sequences=1)
+    
+    # Use max_new_tokens to control the length of the generated output
+    roast = generator(prompt, max_new_tokens=50, num_return_sequences=1)
     
     return roast[0]['generated_text']
 
+
 # Gradio interface function
 def roast_resume(file):
     if file.name.endswith('.pdf'):