Update app.py

app.py

@@ -37,18 +37,13 @@ def CTXGen(X1, X2, τ, g_num, length_range):
     is_stopped = False
     start, end = length_range
 
-    if X1!='X' and X2!='X':
-        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}")]
-        msa = random.choice(msa)
-        X4 = msa.split("|")[3]
-        X5 = msa.split("|")[4]
-        X6 = msa.split("|")[5]
-    else:
-        X4 = ''
-        X5 = ''
-        X6 = ''
+    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}")]
+    msa = random.choice(msa)
+    X4 = msa.split("|")[3]
+    X5 = msa.split("|")[4]
+    X6 = msa.split("|")[5]
 
     model.eval()
     with torch.no_grad():
@@ -153,13 +148,13 @@ def CTXGen(X1, X2, τ, g_num, length_range):
 with gr.Blocks() as demo:
     gr.Markdown("# Conotoxin Generation")
     with gr.Row():
-        X1 = gr.Dropdown(choices=['X','<α7>','<AChBP>','<α4β2>','<α3β4>','<Ca22>','<α3β2>', '<Na12>','<α9α10>','<K16>', '<α1β1γδ>', 
+        X1 = gr.Dropdown(choices=['<α7>','<AChBP>','<α4β2>','<α3β4>','<Ca22>','<α3β2>', '<Na12>','<α9α10>','<K16>', '<α1β1γδ>', 
                                   '<K13>', '<α1BAR>', '<α1β1ε>', '<α1AAR>', '<GluN3A>', '<GluN2B>', '<α75HT3>', '<Na14>', 
                                   '<GluN2C>', '<NET>', '<NavBh>', '<α6β3β4>', '<Na11>', '<Ca13>', '<Ca12>', '<Na16>', '<α6α3β2>', 
                                   '<GluN2A>', '<GluN2D>', '<K17>', '<α1β1δε>', '<GABA>', '<α9>', '<K12>', '<Kshaker>',  '<Na18>', 
                                   '<α6α3β2β3>', '<α1β1δ>', '<α6α3β4β3>', '<α2β2>','<α6β4>', '<α2β4>','<Na13>', '<Na15>', '<α4β4>', 
                                   '<α7α6β2>', '<α1β1γ>', '<NaTTXR>', '<K11>', '<Ca23>', '<α6α3β4>', '<NaTTXS>', '<Na17>'], label="Subtype")
-        X2 = gr.Dropdown(choices=['X','<high>','<low>'], label="Potency")
+        X2 = gr.Dropdown(choices=['<high>','<low>'], label="Potency")
         τ = gr.Slider(minimum=1, maximum=2, step=0.1, label="τ")
         g_num = gr.Dropdown(choices=[1, 10, 20, 30, 40, 50], label="Number of generations")
         length_range = RangeSlider(minimum=8, maximum=50, step=1, value=(12, 16), label="Length range")