Update app.py

app.py

@@ -44,6 +44,7 @@ def CTXGen(τ, g_num, length_range):
     model.eval()
     with torch.no_grad():
         new_seq = None
+        IDs = []
         generated_seqs = []
         generated_seqs_FINAL = []
         cls_pos_all = []
@@ -127,8 +128,9 @@ def CTXGen(τ, g_num, length_range):
                     cls_probability_all.append(cls_probability)
                     act_pos_all.append(act_pos)
                     act_probability_all.append(act_probability)
+                    IDs.append(count+1)
                     out = pd.DataFrame({
-                        'ID':count,
+                        'ID':IDs,
                         'Generated_seq': generated_seqs_FINAL,
                         'Subtype': cls_pos_all,
                         'Subtype_probability': cls_probability_all,