ilyshi commited on
Commit
9c95ebe
1 Parent(s): 8922b9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def print_probs(logits):
21
  sum+=prob
22
  i+=1
23
 
24
- @st.cache
25
  def make_prediction(text):
26
  tokenized_text = tokenizer(text, return_tensors='pt')
27
  with torch.no_grad():
 
21
  sum+=prob
22
  i+=1
23
 
24
+ # @st.cache
25
  def make_prediction(text):
26
  tokenized_text = tokenizer(text, return_tensors='pt')
27
  with torch.no_grad():