Removed CNN

app.py

@@ -61,8 +61,6 @@ def format_as_markdown(predictions: dict) -> str:
 def classify_text(classifier, text, threshold, output_format):
     if classifier == 'Transformer':
         tags = get_tags_bert(bert, text, threshold)
-    elif classifier == 'CNN':
-        tags = []
     elif classifier == 'LSTM':
         tags = get_tags_lstm(lstm, text, threshold)
     else:
@@ -89,7 +87,7 @@ with gr.Blocks(theme="default") as demo:
         # Left column — Input
         with gr.Column(scale=1):
             classifier = gr.Radio(
-                ["Naïve Bayes", "CNN", "LSTM", "Transformer"],
+                ["Naïve Bayes", "LSTM", "Transformer"],
                 value="Transformer",
                 label="Processing model",
                 info="Choose which model to use to process texts",

models/cnn.py

@@ -1,2 +0,0 @@
-def get_tags_cnn(model, text, threshold=0.5):
-    return []