Add references to stormtrooper processor

digitalmethodsinitiative · Oct 11, 2024 · 0bc88a3 · 0bc88a3
1 parent d399921
commit 0bc88a3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/processors/machine_learning/annotate_text.py b/processors/machine_learning/annotate_text.py
@@ -31,6 +31,13 @@ class TextClassifier(BasicProcessor):
                    "provided categories.")  # description displayed in UI
     extension = "csv"  # extension of result file, used internally and in UI
 
+    references = [
+        "Annotations are made using the [Stormtrooper](https://centre-for-humanities-computing.github.io/stormtrooper/) library",
+        "Model card: [google/flan-t5-large](https://huggingface.co/google/flan-t5-large)",
+        "Model card: [tiiuae/falcon-7b-instruct](https://huggingface.co/tiiuae/falcon-7b-instruct)",
+        "Model card: [meta-llama/Meta-Llama-3.1-8B-Instruct](https://huggingface.co/meta-llama/Meta-Llama-3.1-8B-Instruct)"
+    ]
+
     config = {
         "dmi-service-manager.stormtrooper_intro-1": {
             "type": UserInput.OPTION_INFO,