Update app.py

app.py

@@ -166,7 +166,8 @@ m = model.to(device)
 # print(sum(p.numel() for p in m.parameters()) / 1e3, "K parameters")
 
 # load the model.pth
-model.load_state_dict(torch.load("model.pth", weights_only=True))
+model.load_state_dict(torch.load("model.pth", weights_only=false, map_location=torch.device('cpu') )
+
 model.eval()
 
 def respond(