Update app.py

app.py

@@ -144,11 +144,15 @@ def load_model_from_disk(path, model_type):
     model.load_state_dict(state['model_state_dict'] if 'model_state_dict' in state else state)
     model.eval()
     return model
-   
+@st.cache_resource(ttl=3600)
+def load_scripted_model(path):
+    model = torch.jit.load(path, map_location=torch.device("cpu"))
+    model.eval()
+    return model
 @st.cache_resource
 def preload_models():
     return {
-        "Bi-Directional LSTM": load_model_from_disk("best_bilstm_model.pth", model_type="Bi-Directional LSTM"),
+        "Bi-Directional LSTM": load_scripted_model("bilstm_scripted.pt"),
         "Gated Recurrent Unit (GRU)": load_model_from_disk("best_gru_model.pth", model_type="GRU")
     }