Update app.py

app.py

@@ -5,11 +5,7 @@ import torch
 import numpy as np
 import urllib.request
 
-# Initialize session state for memory if not already
-if "memory" not in st.session_state:
-    st.session_state.memory = {"characters": {}, "transcript": ""}
-
-@st.cache_resource
+# Load model without caching due to serialization issue with PretrainedConfig
 def load_model():
     model = AutoModel.from_pretrained("ragavsachdeva/magi", trust_remote_code=True)
     device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -24,6 +20,8 @@ def read_image_as_np_array(image_path):
         image = Image.open(image_path).convert("L").convert("RGB")
     image = np.array(image)
     return image
+    
+model = load_model()    
 
 @st.cache_data
 def predict_detections_and_associations(