Delete modules

modules/dataset.py

@@ -1,16 +0,0 @@
-from datasets import load_dataset
-
-dataset = load_dataset("go_emotions", split="train")
-
-emotions = dataset.info.features["labels"].feature.names
-
-
-def query_emotion(start, end):
-    rows = dataset[start:end]
-
-    observations = [
-        {"text": r[0], "emotion": emotions[r[1][0]]}
-        for r in zip(rows["text"], rows["labels"])
-    ]
-
-    return observations

modules/inference.py

@@ -1,11 +0,0 @@
-from transformers import T5Tokenizer, T5ForConditionalGeneration
-
-tokenizer = T5Tokenizer.from_pretrained("t5-small")
-model = T5ForConditionalGeneration.from_pretrained("t5-small")
-
-
-def infer_t5(input):
-    input_ids = tokenizer(input, return_tensors="pt").input_ids
-    outputs = model.generate(input_ids)
-
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)