Update app.py

app.py

@@ -37,7 +37,7 @@ def load_model():
     return DistilBertForSequenceClassification.from_pretrained('./')
 
 def load_tokenizer():
-    return AutoTokenizer.from_pretrained(base_model_name)
+    return AutoTokenizer.from_pretrained('./')
 
 def top_xx(preds, xx=95):
     tops = torch.argsort(preds, 1, descending=True)
@@ -53,7 +53,7 @@ def top_xx(preds, xx=95):
 
 model = load_model()
 tokenizer = load_tokenizer()
-temperature = 1/2
+temperature = 1
 
 st.title('ArXivTaxonomizer© (original version)')
 st.caption('If you are aware of any other public services which are  illegally providing the ArXivTaxonomizer© functionality, please consider informing us.')
@@ -63,7 +63,7 @@ with st.form("Taxonomizer"):
     title = st.text_area(label='Title', height=30)
     abstract = st.text_area(label='Abstract (optional)', height=200)
     xx = st.slider(label='Verbosity', min_value=1, max_value=99, value=95)
-    st.caption('Lower values will generate a few best guesses. Higher values will lead to a comprehensive list of topics that our model considers relevant. \nEmpirically, values arond 95 work best and generate a list of 3-5 guesses.')
+    st.caption('Lower values will generate a few best guesses. Higher values will lead to a comprehensive list of topics that our model considers relevant. \nEmpirically, values arond 70 work best and generate a list of 3-5 guesses.')
 
     submitted = st.form_submit_button("Taxonomize")
     st.caption('We **do not** recommend using ArXivTaxonomizer© to choose tags for you new paper.')