edit gitignore

.gitignore

@@ -0,0 +1 @@
+__pycache__

app.py

@@ -1,12 +1,14 @@
-import torch
 import streamlit as st
 from random import choice
 from annotated_text import annotated_text
+from resources import *
 from helpers import *
 
-with open("sentences.pt", 'rb') as f:
-    sentences = torch.load(f)
-sentence = choice(sentences)
+base_model = "xlnet-base-cased"
+session = load_variables()
+sentences = load_sentences()
+baseline_classifier = load_model(f"Dagobert42/{base_model}-biored-finetuned")
+augmented_classifier = load_model(f"Dagobert42/{base_model}-biored-augmented")
 
 st.title("Semantic Frame Augmentation")
 st.subheader("Analysing difficult low-resource domains with only a handful of examples")

helpers.py

@@ -1,16 +1,4 @@
-from transformers import pipeline
-
-baseline_classifier = pipeline("ner",
-    model="Dagobert42/xlnet-base-cased-biored-augmented",
-    aggregation_strategy="average"
-    )
-augmented_classifier = pipeline("ner",
-    model="Dagobert42/xlnet-base-cased-biored-finetuned",
-    aggregation_strategy="average"
-    )
-
-def annotate_sentence(sentence, predictions):
-    colors = {
+colors = {
         'GeneOrGeneProduct': '#aad4aa',  # Pastel green
         'DiseaseOrPhenotypicFeature': '#f8b400',  # Pastel orange
         'ChemicalEntity': '#a4c2f4',  # Pastel blue