Update app.py

app.py

@@ -24,6 +24,12 @@ def stop_generation():
     return "Generation stopped."
 
 def CTXGen(X0, X1, X2, τ, g_num, model_name):
+    print(X0)
+    print(X1)
+    print(X2)
+    print(τ)
+    print(g_num)
+    print(model_name)
     global is_stopped
     is_stopped = False
 
@@ -35,7 +41,7 @@ def CTXGen(X0, X1, X2, τ, g_num, model_name):
     train_seq = train_seqs['Seq'].tolist()
     model = torch.load(save_path, map_location=torch.device('cpu'))
     model = model.to(device)
-    
+    print(model)
 
     X3 = "X" * len(X0)
     msa_data = pd.read_csv('conoData_C0.csv') 
@@ -96,6 +102,7 @@ def CTXGen(X0, X1, X2, τ, g_num, model_name):
                 break
 
             seq = [f"{X1}|{X2}|{X3}|{X4}|{X5}|{X6}"]
+            print(seq)
             vocab_mlm.token_to_idx["X"] = 4
 
             padded_seq, _, _, _ = get_paded_token_idx_gen(vocab_mlm, seq, new_seq)