Update app.py

app.py

@@ -48,7 +48,7 @@ class MyModel(nn.Module):
         return torch.softmax(output_feature,dim=1)
 
 model = MyModel()
-model.load_state_dict(torch.load("best_model.pth"))
+model.load_state_dict(torch.load("best_model.pth", map_location=torch.device('cpu')))
 model = model.to(device)
 model.eval()
 
@@ -70,7 +70,7 @@ def conotoxinfinder(files):
     fr=open(files, 'r')
     seqs = []
     for line in fr:
-        if not line.startswith('>'):    #判断字符串是否以‘>开始’
+        if not line.startswith('>'):    
             seqs.append(line)
     seq_all = []
     output_all = []