Update app.py

app.py

@@ -10,6 +10,9 @@ model = GemmaForCausalLM.from_pretrained('google/gemma-2-9b',
                                         )
 # pipe = pipeline('text-generation', model=model,tokenizer = tokenizer)
 
+MAX_MAX_NEW_TOKENS = 2048
+DEFAULT_MAX_NEW_TOKENS = 1024
+
 @spaces.GPU(duration=120)
 def generate(
     message: str,