Update app.py

app.py

@@ -1,3 +1,29 @@
+import random
+import torch
+import gradio as gr
+from gradio_rangeslider import RangeSlider
+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,100000)
+
+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'  
+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):
+    logits = logits / temperature
+    probabilities = torch.softmax(logits, dim=-1)
+    sampled_token = torch.multinomial(probabilities, 1)
+    return sampled_token
+
 def CTXGen(τ, g_num, length_range, progress=gr.Progress()):
     start, end = length_range
     X1 = "X"