Update app.py

app.py

@@ -66,7 +66,7 @@ class NewsClassifier(nn.Module):
         x = self.fc(x)
         return x
 
-@st.cache_data
+@st.cache
 def load_models(model_type):
     models = None
     model = None