Update app.py

app.py

@@ -6,12 +6,19 @@ model_name = "gpt2"
 GPT2Tokenizer.from_pretrained(model_name)
 GPT2LMHeadModel.from_pretrained(model_name)
 
-# Load the pipeline
-generator = pipeline('text-generation', model=model_name)
+# Load the pipeline with additional parameters
+generator = pipeline(
+    'text-generation',
+    model=model_name,
+    do_sample=True,  # Enables sampling for more creative output
+    temperature=0.7,  # Controls randomness (0.7 is balanced for creativity and coherence)
+    top_k=50,  # Limits the vocabulary to the top 50 most likely tokens
+    max_length=150  # Increased length for longer responses
+)
 
 # Function to generate text
 def generate_text(prompt):
-    output = generator(prompt, max_length=50, num_return_sequences=1)
+    output = generator(prompt, num_return_sequences=1)
     return output[0]['generated_text']
 
 # Build the interface
@@ -23,5 +30,5 @@ iface = gr.Interface(
     description="Generate text with AI!"
 )
 
-# Launch it kk
+# Launch it
 iface.launch()