Spaces:

olya-const
/

lab2

Runtime error

olya-const commited on Mar 29, 2022

Commit

0ea3763

•

1 Parent(s): 88b4df7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,8 @@ import datasets
 def load_model():
      return AutoModelForSequenceClassification.from_pretrained('./')
-tokenizer = AutoTokenizer.from_pretrained('distilbert-base-cased')
 model = load_model()
 title = st.text_area('Title')

 def load_model():
      return AutoModelForSequenceClassification.from_pretrained('./')
+if 'tokenizer' not in globals():
+    tokenizer = AutoTokenizer.from_pretrained('distilbert-base-cased')
 model = load_model()
 title = st.text_area('Title')