Spaces:
Runtime error
Runtime error
Jeffrey Rathgeber Jr
commited on
test
Browse files
app.py
CHANGED
@@ -72,6 +72,7 @@ if option == 'MILESTONE 3: FINE-TUNED':
|
|
72 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
73 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
74 |
classifier = pipeline(task="sentiment-analysis", model=model, tokenizer=tokenizer)
|
|
|
75 |
# pipeline
|
76 |
preds = classifier(textIn)
|
77 |
preds = [{"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds]
|
|
|
72 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
73 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
74 |
classifier = pipeline(task="sentiment-analysis", model=model, tokenizer=tokenizer)
|
75 |
+
|
76 |
# pipeline
|
77 |
preds = classifier(textIn)
|
78 |
preds = [{"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds]
|