Update app.py

app.py

@@ -14,7 +14,8 @@ def generate_text(prompt):
     model = PeftModel.from_pretrained(base_model, "diabolic6045/gemma-2-2b-chess-adapter")
     model.tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-2b")
     input_ids = model.tokenizer.encode(prompt, return_tensors="pt")
-    output = model.generate(input_ids, max_length=100)
+    print("generation")
+    output = model.generate(input_ids, max_length=20)
     return model.tokenizer.decode(output[0], skip_special_tokens=True)
 
 # Create a Gradio interface