more broken virtualenvs

language-models-project/language_models_project/main.py

@@ -0,0 +1,8 @@
+from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
+def classify():
+    tokenizer = AutoTokenizer.from_pretrained("finiteautomata/bertweet-base-sentiment-analysis")
+    model = AutoModelForSequenceClassification.from_pretrained("finiteautomata/bertweet-base-sentiment-analysis")
+    sentiment_pipeline = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
+    return sentiment_pipeline

language-models-project/tests/test_classifier.py

@@ -0,0 +1,6 @@
+from language_models_project.main import classify
+
+def testclassifier():
+    """ Tests that the classifier classifies as expected """
+    test_cases = ["This restaurant is awesome", "This restaurant is awful"]
+