Jeffrey Rathgeber Jr commited on
Commit
c76cf19
·
unverified ·
1 Parent(s): b9ff321
Files changed (1) hide show
  1. app.py +1 -0
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]