Fix

app.py

@@ -26,7 +26,7 @@ m_30_40_model = AutoModelForSequenceClassification.from_pretrained(m_30_40_model
 m_40_55_model_path = "models/MALE_40_55model"
 m_40_55_model = AutoModelForSequenceClassification.from_pretrained(m_40_55_model_path)
 
-def classify_text(test_text, selected_model):
+def classify_text(test_text, selected_model='Model 3'):
 
     categories = ['Kvinner 30-40', 'Kvinner 40-55', 'Menn 30-40', 'Menn 40-55']
     
@@ -92,8 +92,8 @@ examples_2 = [
             ["Title: 2023 er det varmeste året noen gang registrert, Text lead: En ny rapport viser at 2023 er det varmeste året registrert siden man startet målingene. Klimaforsker kaller tallene urovekkende."]
 ]
 
-io1 = gr.Interface(fn=classify_text, inputs=["text", gr.Dropdown(['Model 1', 'Model 2'])], outputs='text',  examples=examples_1, title=app_title)
-io2 = gr.Interface(fn=classify_text, inputs=["text", 'Model 3'], outputs='text',  examples=examples_2, title=app_title)
+io1 = gr.Interface(fn=classify_text, inputs=["text", gr.Dropdown(['Model 1', 'Model 2'])], outputs='label',  examples=examples_1, title=app_title)
+io2 = gr.Interface(fn=classify_text, inputs=["text"], outputs='text',  examples=examples_2, title=app_title)
 
 gr.TabbedInterface(
     [io1, io2], ["Model 1 & 2", "Model 3"]