add lab model

Add lab model spaces

app.py

@@ -20,6 +20,8 @@ import base64
 import json
 import os
 import re
+import transformers
+from transformers import pipeline
 
 @st.cache_resource
 def load_sp():
@@ -146,12 +148,19 @@ def load_model(model_name, file_name):
     model = fasttext.load_model(model_path)
     return model
 
+@st.cache_resource
+def load_model_pipeline(model_name, file_name):
+    model = pipeline("text-classification", model=model_name)
+    return model
+    
+
 
 model_1 = load_model(constants.MODEL_NAME, "model_v1.bin")
 model_2 = load_model(constants.MODEL_NAME, "model_v2.bin")
 model_3 = load_model(constants.MODEL_NAME, "model_v3.bin")
 openlid = load_model('laurievb/OpenLID', "model.bin")
 nllb = load_model('facebook/fasttext-language-identification', "model.bin")
+za_lid = load_model('dsfsi/za-lid-bert', "model.bin")
 
 
 # @st.cache_resource
@@ -196,8 +205,10 @@ def compute(sentences, version = 'v3'):
         model_choice = model_3   
     elif version == 'v2':
         model_choice = model_2
-    else:
+    elif:
         model_choice = model_1
+    else:
+        model_choice = za_lid
         
     my_bar = st.progress(0, text=progress_text)
 
@@ -257,8 +268,8 @@ with tab1:
 
     version = st.radio(
         "Choose model",
-        ["nllb-218", "openlid-201", "v1", "v2", "v3"],
-        captions=["NLLB", "OpenLID", "GlotLID version 1", "GlotLID version 2", "GlotLID version 3 (More languages, better quality data)"],
+        ["nllb-218", "openlid-201", "v1", "v2", "v3", "Za_LID"],
+        captions=["NLLB", "OpenLID", "GlotLID version 1", "GlotLID version 2", "GlotLID version 3 (More languages, better quality data)", "za BERT"],
         index = 4,
         key = 'version_tab1',
         horizontal = True
@@ -296,8 +307,8 @@ with tab2:
 
     version = st.radio(
         "Choose model",
-        ["nllb-218", "openlid-201", "v1", "v2", "v3"],
-        captions=["NLLB", "OpenLID", "GlotLID version 1", "GlotLID version 2 (more data and languages), GlotLID version 3 (More languages, better quality data)"],
+        ["nllb-218", "openlid-201", "v1", "v2", "v3", "Za_LID"],
+        captions=["NLLB", "OpenLID", "GlotLID version 1", "GlotLID version 2 (more data and languages), GlotLID version 3 (More languages, better quality data)", "ZA BERT"],
         index = 4,
         key = 'version_tab2',
         horizontal = True