diff --git a/README.md b/README.md
index 7ef2f3f..2768234 100644
--- a/README.md
+++ b/README.md
@@ -79,7 +79,7 @@ print(message)  # message is a binary vector of 16 bits
 # To detect the messages in the low-level.
 result, message = detector(watermarked_audio)
 
-# result is a tensor of size batch x 2 x frames, indicating the probablity (positive and negative) of watermarking for each frame
+# result is a tensor of size batch x 2 x frames, indicating the probability (positive and negative) of watermarking for each frame
 # A watermarked audio should have result[:, 1, :] > 0.5
 print(result[:, 1 , :])  
 
diff --git a/generator.pth b/generator.pth
deleted file mode 100644
index e69de29..0000000
diff --git a/generator.pth.1 b/generator.pth.1
deleted file mode 100644
index 1fc4432..0000000
Binary files a/generator.pth.1 and /dev/null differ