Update app.py

app.py

@@ -49,7 +49,7 @@ def Kmers_funct(seq,num):
 
 def ACE(file):
     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')), strict=False)
     model = model.to(device)
     model.eval()
     tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)