Update app.py

app.py

@@ -69,12 +69,6 @@ st.markdown(
 )
 
 fasta_input = st.text_area("Insere a sequência FASTA:", height=200)
-selected_model = st.selectbox("Modelo a utilizar:", [
-    "ProtBERT (MLP)",
-    "ProtBERT-BFD (MLP)",
-    "ESM-2 (MLP)",
-    "Ensemble (Stacking)"
-])
 predict_clicked = st.button("Prever GO terms")
 
 if predict_clicked:
@@ -107,7 +101,7 @@ if predict_clicked:
 
     # ——— 4) RESULTADOS ———
     def mostrar(tag, y_pred):
-        with st.expander(tag, expanded=True):
+        with st.expander(tag, expanded=(tag == "Ensemble (Stacking)")):
             hits = mlb.inverse_transform((y_pred >= THRESH).astype(int))[0]
             st.markdown(f"**GO terms com prob ≥ {THRESH}**")
             if hits:
@@ -124,13 +118,9 @@ if predict_clicked:
                 name, _ = GO_INFO.get(go_id, ("", ""))
                 st.write(f"{go_id} — {name} : {y_pred[0][idx]:.4f}")
 
-    # ——— 5) MOSTRAR RESULTADO DO MODELO ESCOLHIDO ———
-    if selected_model == "ProtBERT (MLP)":
-        mostrar("ProtBERT (MLP)", y_pb)
-    elif selected_model == "ProtBERT-BFD (MLP)":
-        mostrar("ProtBERT-BFD (MLP)", y_bfd)
-    elif selected_model == "ESM-2 (MLP)":
-        mostrar("ESM-2 (MLP)", y_esm)
-    else:
-        mostrar("Ensemble (Stacking)", y_ens)
-
+    # ——— MOSTRAR APENAS O ENSEMBLE POR DEFEITO ———
+    
+    # mostrar("ProtBERT (MLP)",      y_pb)
+    # mostrar("ProtBERT-BFD (MLP)",  y_bfd)
+    # mostrar("ESM-2 (MLP)",         y_esm)
+    mostrar("Ensemble (Stacking)", y_ens)