app

app.py

@@ -478,13 +478,13 @@ def predict_raw(raw_input):
     state_dict = torch.load(modelfile, map_location=torch.device(device))
     new_state_dict = OrderedDict()
 
-	for k, v in state_dict.items():
-	    name = k.replace('module.','')
-	    new_state_dict[name] = v
+    for k, v in state_dict.items():
+        name = k.replace('module.','')
+        new_state_dict[name] = v
 
-	model = CNN_linear().to(device)
-	model.load_state_dict(new_state_dict, strict = False)
-	model.eval()
+    model = CNN_linear().to(device)
+    model.load_state_dict(new_state_dict, strict = False)
+    model.eval()
 
     # st.write('====Parse Input====')
     ids, seqs = read_raw(raw_input)
@@ -492,11 +492,11 @@ def predict_raw(raw_input):
     # st.write('====Predict====')
     res_pd = pd.DataFrame()
     for wt_seq, wt_id in zip(seqs, ids):
-    	try:
-	    	res = mutated_seq(wt_seq, wt_id)
-			res_pd.append(res)
-		except:
-			st.write('====Please Try Again this sequence: ', wt_id, wt_seq)
+        try:
+            res = mutated_seq(wt_seq, wt_id)
+            res_pd.append(res)
+        except:
+            st.write('====Please Try Again this sequence: ', wt_id, wt_seq)
     # print(pred)
     return res_pd