Spaces:
Sleeping
Sleeping
Update pytorch_test.py
Browse files- 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,
|
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',
|