Update app.py

app.py

@@ -40,8 +40,8 @@ def chat_with_distilgpt2(input_text, temperature, top_p, top_k):
     # Generate the response using the model with adjusted parameters
     outputs = model.generate(
         input_ids,
-        max_length=input_ids.shape[1] + 100,  # Limit total length
-        max_new_tokens=100,
+        max_length=input_ids.shape[1] + 50,  # Limit total length
+        max_new_tokens=15,
         num_return_sequences=1,
         no_repeat_ngram_size=3,
         repetition_penalty=1.2,