Hugging Face's logo

Spaces:
RayCappola
/
YSDA_homework_nlp_ops
Runtime error

App Files Community
RayCappola commited on
Commit
f0930e7
•
1 Parent(s): e731bb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ labels_articles = {1: 'Computer Science',2: 'Economics',3: "Electrical Engineeri
40
 
41
  4: "Mathematics",5: "Physics",6: "Quantitative Biology",7: "Quantitative Finance", 8: "Statistics"}
42
 
43
- @st.cache
44
  def load_models():
45
  model=Net()
46
  model.load_state_dict(torch.load('dummy_model.txt', map_location=torch.device('cpu')))
 
40
 
41
  4: "Mathematics",5: "Physics",6: "Quantitative Biology",7: "Quantitative Finance", 8: "Statistics"}
42
 
43
+ @st.cache(allow_output_mutation=True)
44
  def load_models():
45
  model=Net()
46
  model.load_state_dict(torch.load('dummy_model.txt', map_location=torch.device('cpu')))