Update app.py

app.py

@@ -9,7 +9,7 @@ pretrained_vit_path = 'pretrained_vit_model_full.pth'
 class_names = ['Angry', 'Other', 'Sad', 'happy']
 
 # Load your model
-@st.cache(allow_output_mutation=True)
+@st.experimental_singleton
 def load_model():
     model = torch.load(pretrained_vit_path, map_location=torch.device('cpu'))
     model.eval()