Update app.py

app.py

@@ -5,7 +5,7 @@ import pandas as pd
 from utils import create_vocab, setup_seed
 from dataset_mlm import  get_paded_token_idx_gen, add_tokens_to_vocab
 import time 
-seed = random.randint(0,99999999)
+seed = random.randint(0,100000)
 
 setup_seed(seed)
 device = torch.device("cpu")
@@ -14,7 +14,7 @@ vocab_mlm = add_tokens_to_vocab(vocab_mlm)
 save_path = 'mlm-model-27.pt'  #1
 train_seqs = pd.read_csv('C0_seq.csv')  #2
 train_seq = train_seqs['Seq'].tolist()
-model = torch.load(save_path, weights_only=False, map_location=torch.device('cpu'))
+model = torch.load(save_path, map_location=torch.device('cpu'))
 model = model.to(device)
 
 def temperature_sampling(logits, temperature):