Cipher-AI
/

Substitution-Cipher-Alphabet-Eng

Text2Text Generation

text-generation-inference

Model card Files Files and versions Community

suayptalha commited on 29 days ago

Commit

b524a09

·

verified ·

1 Parent(s): 9a23d03

Update README.md

Files changed (1) hide show

README.md +2 -4

README.md CHANGED Viewed

@@ -49,12 +49,10 @@ from string import ascii_lowercase
 import Levenshtein
 import random
-tokenizer = AutoTokenizer.from_pretrained("Cipher-AI/Substitution-Cipher-Alphabet-Eng")
-model = AutoModelForSeq2SeqLM.from_pretrained("Cipher-AI/Substitution-Cipher-Alphabet-Eng")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-alphabet_model = model.to(device)
 correction_model = AutoModelForSeq2SeqLM.from_pretrained("Cipher-AI/AutoCorrect-EN-v2").to(device)
 def similarity_percentage(s1, s2):

 import Levenshtein
 import random
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+tokenizer = AutoTokenizer.from_pretrained("Cipher-AI/Substitution-Cipher-Alphabet-Eng")
+alphabet_model = AutoModelForSeq2SeqLM.from_pretrained("Cipher-AI/Substitution-Cipher-Alphabet-Eng").to(device)
 correction_model = AutoModelForSeq2SeqLM.from_pretrained("Cipher-AI/AutoCorrect-EN-v2").to(device)
 def similarity_percentage(s1, s2):