Update app.py

app.py

@@ -39,7 +39,7 @@ def conotoxinfinder(files):
                 seqs.append(line)
 
     model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint, num_labels=1)
-    model.load_state_dict(torch.load("best_model.pth"))
+    model.load_state_dict(torch.load("best_model.pth", map_location=torch.device('cpu')), strict=False)
     model = model.to(device)
 
     value_all = []