Update app.py

app.py

@@ -23,12 +23,11 @@ def initialize_model(hf_token):
             device_map="auto"  # This will automatically handle GPU if available
         )
         
-        # Configure watermarking
+        # Configure watermarking with only the supported parameters
         WATERMARK_KEYS = [654, 400, 836, 123, 340, 443, 597, 160, 57, 789]
         watermarking_config = SynthIDTextWatermarkingConfig(
             keys=WATERMARK_KEYS,
-            ngram_len=5,
-            gamma=0.5,
+            ngram_len=5
         )
         
         return model, tokenizer, watermarking_config, "Model initialized successfully!"