From 6dc6fd8bdf69d5f7f9f5a5ae488ae1eba6d546bd Mon Sep 17 00:00:00 2001
From: marwan2232004 <118024824+marwan2232004@users.noreply.github.com>
Date: Sat, 12 Oct 2024 10:54:45 +0300
Subject: [PATCH] Fix: Deployment device problem

---
 Inference.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Inference.py b/Inference.py
index a844d0e..8f0965e 100644
--- a/Inference.py
+++ b/Inference.py
@@ -47,7 +47,7 @@ def predict(audio_file):
         dim_feedforward=2048,
     )
 
-    model.load_state_dict(torch.load(model_path, weights_only=True))
+    model.load_state_dict(torch.load(model_path, weights_only=False))
     model.to(device)
     model.eval()