Update app.py

app.py

@@ -3,8 +3,8 @@ from transformers import pipeline
 import streamlit as st
 import streamlit.components.v1 as components
 
-pipe_1 = pipeline("text-classification", model="mavinsao/mi-roberta-base-finetuned-classification")
-pipe_2 = pipeline("text-classification", model="mavinsao/roberta-mental-finetuned")
+pipe_1 = pipeline("text-classification", model="mavinsao/roberta-base-finetuned-mental-health")
+pipe_2 = pipeline("text-classification", model="mavinsao/mi-roberta-base-finetuned-mental-health")
 
 
 # Streamlit app with background image
@@ -28,6 +28,7 @@ def ensemble_predict(text):
     results_2 = pipe_2(text)
 
     ensemble_scores = {} 
+    
     for results in [results_1, results_2]:  # Iterate through predictions
         for result in results:
             label = result['label']