oucgc1996 commited on
Commit
1c09c7c
·
verified ·
1 Parent(s): 81be004

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(), "output.csv"
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(), "output.csv"
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)