Update app.py
Browse files
app.py
CHANGED
@@ -65,7 +65,7 @@ def CTXGen(X1, X2, τ, g_num, length_range, model_name):
|
|
65 |
while count < gen_num:
|
66 |
new_seq = None
|
67 |
if is_stopped:
|
68 |
-
return pd.DataFrame()
|
69 |
|
70 |
if time.time() - start_time > 1200:
|
71 |
break
|
@@ -83,7 +83,7 @@ def CTXGen(X1, X2, τ, g_num, length_range, model_name):
|
|
83 |
|
84 |
for i in range(length):
|
85 |
if is_stopped:
|
86 |
-
return pd.DataFrame()
|
87 |
|
88 |
_, idx_seq, idx_msa, attn_idx = get_paded_token_idx_gen(vocab_mlm, seq, new_seq)
|
89 |
idx_seq = torch.tensor(idx_seq).unsqueeze(0).to(device)
|
|
|
65 |
while count < gen_num:
|
66 |
new_seq = None
|
67 |
if is_stopped:
|
68 |
+
return "output.csv", pd.DataFrame()
|
69 |
|
70 |
if time.time() - start_time > 1200:
|
71 |
break
|
|
|
83 |
|
84 |
for i in range(length):
|
85 |
if is_stopped:
|
86 |
+
return "output.csv", pd.DataFrame()
|
87 |
|
88 |
_, idx_seq, idx_msa, attn_idx = get_paded_token_idx_gen(vocab_mlm, seq, new_seq)
|
89 |
idx_seq = torch.tensor(idx_seq).unsqueeze(0).to(device)
|