Spaces:

oucgc1996
/

CreoPep_Unconstrained_generation

Running

oucgc1996 commited on Mar 10

Commit

d1daeb5

verified ·

1 Parent(s): df1eb63

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,8 +12,8 @@ setup_seed(seed)
 device = torch.device("cpu")
 vocab_mlm = create_vocab()
 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, map_location=torch.device('cpu'))
 model = model.to(device)

 device = torch.device("cpu")
 vocab_mlm = create_vocab()
 vocab_mlm = add_tokens_to_vocab(vocab_mlm)
+save_path = 'mlm-model-27.pt'
+train_seqs = pd.read_csv('C0_seq.csv')
 train_seq = train_seqs['Seq'].tolist()
 model = torch.load(save_path, map_location=torch.device('cpu'))
 model = model.to(device)