Update app.py

app.py

@@ -36,10 +36,10 @@ if not st.session_state.models:
 
     # models
     models = {  
-        "SVM": SVC(kernel='linear'),  
+        # "SVM": SVC(kernel='linear'),  
         "Logistic Regression": LogisticRegression(max_iter=1000),  
         "Random Forest": RandomForestClassifier(n_estimators=10),
-        "Gradient Boosting": GradientBoostingClassifier()  
+        # "Gradient Boosting": GradientBoostingClassifier()  
     }  
 
     for name, model in models.items():  
@@ -82,9 +82,11 @@ if not st.session_state.models:
     # st.session_state.reports["BERT"] = pd.DataFrame(report).transpose()  
 
 if st.session_state.accuracy:  
-    
+    color = ['blue','orange']
     plt.figure(figsize=(10, 5))  
-    plt.bar(st.session_state.accuracy.keys(), st.session_state.accuracy.values(), color=['blue', 'orange', 'green','red', 'purple'])  
+    # plt.bar(st.session_state.accuracy.keys(), st.session_state.accuracy.values(), color=['blue', 'orange', 'green','red', 'purple'])  
+    plt.bar(st.session_state.accuracy.keys(), st.session_state.accuracy.values(), color=color)  
+    
     plt.ylabel('Accuracy')  
     plt.title('Model Accuracy Comparison')  
     st.pyplot(plt)