Update app.py

app.py

@@ -9,6 +9,8 @@ labels_articles = {1: 'Computer Science', 2: 'Economics', 3: "Electrical Enginee
                    4: "Mathematics", 5: "Physics", 6: "Quantitative Biology", 7: "Quantitative Finance", 
                    8: "Statistics"
                   }
+                  
+@st.cache(allow_output_mutation=True)
 def models():
     class Net(nn.Module):
         def __init__(self):