fix error

app.py

@@ -68,7 +68,7 @@ mi_model.load_state_dict(mi_loaded_model_state['state_dict'])
 
 
 
-def predict_label(sentence, tokenizer, model, device):
+def predict_label(sentence, tokenizer, model1, model2, device):
     # Tokenize the sentence and create attention mask
     tokenized_input = tokenizer(
         sentence,
@@ -121,5 +121,5 @@ sentence = st.text_area("Enter the sentence to predict your mental illness state
 # Prediction button
 if st.button('Predict'):
     # Predict label
-    predicted_response = predict_label(sentence, tokenizer, model, device)
+    predicted_response = predict_label(sentence, tokenizer, mi_model, roberta_model, device)
     st.json(predicted_response)