Update app.py

app.py

@@ -3,8 +3,8 @@ from transformers import pipeline
 
 generator = pipeline(
     "text-generation",
-    model="gpt2",
-    pad_token_id=50256  # crucial for GPT-2
+    model="distilgpt2",
+    pad_token_id=50256  # DistilGPT2 uses the same special tokens as GPT2
 )
 
 def generate_cv(name, education, experience):
@@ -16,7 +16,13 @@ def generate_cv(name, education, experience):
         "CV:\n"
     )
     # Use max_new_tokens=100 (or something smaller) for speed
-    outputs = generator(prompt, max_new_tokens=100)
+outputs = generator(
+    prompt, 
+    max_new_tokens=50,
+    do_sample=True,
+    top_k=50,
+    temperature=0.7
+)
     return outputs[0]["generated_text"]
 
 demo = gr.Interface(