jnhdny commited on
Commit
1392f01
1 Parent(s): 4fc35c6

Fix rpw implementation

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -134,6 +134,8 @@ class WatermarkDetector(WatermarkBase):
134
 
135
  if self.seeding_scheme == "simple_1":
136
  self.min_prefix_len = 1
 
 
137
  else:
138
  raise NotImplementedError(f"Unexpected seeding_scheme: {self.seeding_scheme}")
139
 
 
134
 
135
  if self.seeding_scheme == "simple_1":
136
  self.min_prefix_len = 1
137
+ elif self.seeding_scheme == "rpw":
138
+ self.min_prefix_len = 2
139
  else:
140
  raise NotImplementedError(f"Unexpected seeding_scheme: {self.seeding_scheme}")
141