Update app.py

app.py

@@ -9,8 +9,10 @@ import matplotlib.pyplot as plt
 import numpy as np
 import seaborn as sns
 # Load the model
-
-model = joblib.load('85pct(new).pkl')
+def read_model(region):
+    if(region == "United States"):
+        model = joblib.load('85pct(2).pkl')
+    return model
 
 # Define the categories
 categories = {
@@ -114,6 +116,8 @@ def main():
 
     # Input fields
     if selected_option == "Predict":
+        region = st.sidebar.selectbox("Select Region", ['United States'])
+        model = read_model(region)
         with st.container():
             col1, col2, col3 = st.columns(3)
             getTitle, getDuration, getCategory = "", 0.00, 1
@@ -161,7 +165,7 @@ def main():
                     st.warning("Please enter a duration.")
 
                 else:
-                    prediction = predict_trend(title, duration, categoryId)
+                    prediction = predict_trend(model,title, duration, categoryId)
                     if prediction[0] == 1:
                         st.success("This video is predicted to be a trend!")
                         st.markdown("![Alt Text](https://media.tenor.com/Cyi2zT7wcmcAAAAj/pentol-gif-eak.gif)")
@@ -182,7 +186,7 @@ def main():
                 st.download_button(
                     "Download",
                     csv,
-                    "file.csv",
+                    "top10Trending.csv",
                     "text/csv",
                     key='download-csv'
                 )
@@ -241,7 +245,7 @@ def main():
                     show_top_titleLength()
                 with col13:
                     show_best_titleLength()
-                    
+         
 def get_picture_from_url(url):
     try:
         response = requests.get(url)
@@ -388,7 +392,7 @@ def show_best_titleLength():
         st.write(f"<span style='color:{color};font-weight:bold;'>{range}</span>", unsafe_allow_html=True)
 
 # Function to make predictions
-def predict_trend(title, duration, category_id):
+def predict_trend(model,title, duration, category_id):
     duration = str(duration)
     category_id = int(category_id)
     clean_new_title = preprocess(title)