Spaces:
Runtime error
Runtime error
Commit
·
8cea7b8
1
Parent(s):
0225db4
Update main.py
Browse files
main.py
CHANGED
@@ -3,6 +3,9 @@ import pandas as pd
|
|
3 |
import transformers as pipeline
|
4 |
from transformers import AutoTokenizer,AutoModelForSequenceClassification
|
5 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
|
|
|
|
|
|
6 |
|
7 |
model_name = "Sonny4Sonnix/Movie_Sentiments_Analysis_with_FastAPI" # Replace with the name of the pre-trained model you want to use
|
8 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
|
|
3 |
import transformers as pipeline
|
4 |
from transformers import AutoTokenizer,AutoModelForSequenceClassification
|
5 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
6 |
+
import os
|
7 |
+
os.environ["TRANSFORMERS_CACHE"] = "/path/to/writable/cache/directory"
|
8 |
+
|
9 |
|
10 |
model_name = "Sonny4Sonnix/Movie_Sentiments_Analysis_with_FastAPI" # Replace with the name of the pre-trained model you want to use
|
11 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|