api token

app.py

@@ -1,3 +1,4 @@
+import os
 import streamlit as st
 import transformers
 import torch
@@ -8,9 +9,9 @@ model_name = st.selectbox(
     'Выберите модель',
     ('tinkoff-ai/crossencoder-tiny', 'tinkoff-ai/crossencoder-medium', 'tinkoff-ai/crossencoder-large')
 )
-
-tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
-model = transformers.AutoModelForSequenceClassification.from_pretrained(model_name)
+auth_token = os.environ.get('TOKEN') or True
+tokenizer = transformers.AutoTokenizer.from_pretrained(model_name, use_auth_token=auth_token)
+model = transformers.AutoModelForSequenceClassification.from_pretrained(model_name, use_auth_token=auth_token)
 if torch.cuda.is_available():
     model = model.cuda()