gourib commited on
Commit
04f6f70
1 Parent(s): c693553

Update pytorch_test.py

Browse files
Files changed (1) hide show
  1. pytorch_test.py +1 -1
pytorch_test.py CHANGED
@@ -44,7 +44,7 @@ def run_model(input_text) :
44
  #process.env.HF_ACCESS_TOKEN = HF_TOKEN
45
  #os.environ["HF_ACCESS_TOKEN"] = HF_TOKEN
46
  login(token = HF_TOKEN)
47
- model = AutoModelForSequenceClassification.from_pretrained(model_name, use_auth_token=True)
48
  tokenizer = AutoTokenizer.from_pretrained(model_name)
49
  text_Completion = pipeline(
50
  'text-generation', #'sentiment-analysis',
 
44
  #process.env.HF_ACCESS_TOKEN = HF_TOKEN
45
  #os.environ["HF_ACCESS_TOKEN"] = HF_TOKEN
46
  login(token = HF_TOKEN)
47
+ model = AutoModelForSequenceClassification.from_pretrained(model_name, token=HF_TOKEN)
48
  tokenizer = AutoTokenizer.from_pretrained(model_name)
49
  text_Completion = pipeline(
50
  'text-generation', #'sentiment-analysis',