Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 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)
 
 setup_seed(seed)
@@ -48,7 +48,10 @@ def CTXGen(τ, g_num, start, end):
                         '<Na13>', '<Na12>', '<Na15>', '<α4β4>', '<α7α6β2>', '<α1β1γ>', '<NaTTXR>', '<K11>', '<Ca23>', 
                         '<α9α10>','<α6α3β4>', '<NaTTXS>', '<Na17>','<high>','<low>','[UNK]','[SEP]','[PAD]','[CLS]','[MASK]']
 
+        start_time = time.time()
         while count < gen_num:
+            if time.time() - start_time > 60:
+                break
             gen_len = random.randint(int(start), int(end))
             X3 = "X" * gen_len
             seq = [f"{X1}|{X2}|{X3}|{X4}|{X5}|{X6}"]