Spaces:

oucgc1996
/

CreoPep_Unconstrained_generation

Running

oucgc1996 commited on Mar 10

Commit

304df7d

verified ·

1 Parent(s): d1daeb5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ 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)
 def temperature_sampling(logits, temperature):

 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)
 model = model.to(device)
 def temperature_sampling(logits, temperature):