Update app.py

app.py

@@ -24,6 +24,9 @@ def stop_generation():
     return "Generation stopped."
 
 def CTXGen(X0, X1, X2, τ, g_num, model_name):
+    global is_stopped
+    is_stopped = False
+
     device = torch.device("cpu")
     vocab_mlm = create_vocab()
     vocab_mlm = add_tokens_to_vocab(vocab_mlm)
@@ -33,9 +36,9 @@ def CTXGen(X0, X1, X2, τ, g_num, model_name):
     model = torch.load(save_path, map_location=torch.device('cpu'))
     model = model.to(device)
     
-    global is_stopped
-    is_stopped = False
+
     X3 = "X" * len(X0)
+    print(X3)
     msa_data = pd.read_csv('conoData_C0.csv') 
     msa = msa_data['Sequences'].tolist()
     msa = [x for x in msa if x.startswith(f"{X1}|{X2}")]
@@ -93,7 +96,6 @@ def CTXGen(X0, X1, X2, τ, g_num, model_name):
             if time.time() - start_time > 1200:
                 break
 
-            gen_len = len(X0)
             seq = [f"{X1}|{X2}|{X3}|{X4}|{X5}|{X6}"]
             vocab_mlm.token_to_idx["X"] = 4