Remove cache optimization (doesn't work for now)

app.py

@@ -15,16 +15,11 @@ gc.enable()
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
-@st.cache_resource
-def load_model():
-    with st.spinner("Loading PointNet++ model..."):
-        checkpoint = torch.load('checkpoints/best_model.pth', map_location=torch.device(device))
-        classifier = pn2.get_model(num_class=4, normal_channel=False)
-        classifier.load_state_dict(checkpoint['model_state_dict'])
-        classifier.eval()
-    return classifier
-
-classifier = load_model()
+with st.spinner("Loading PointNet++ model..."):
+    checkpoint = torch.load('checkpoints/best_model.pth', map_location=torch.device(device))
+    classifier = pn2.get_model(num_class=4, normal_channel=False)
+    classifier.load_state_dict(checkpoint['model_state_dict'])
+    classifier.eval()
 
 side_bg = "static/sidebar.png"
 side_bg_ext = "png"