Update app.py

app.py

@@ -1,24 +1,18 @@
-import gradio as gr
+import gdown
 import joblib
 import requests
 from gensim.models import KeyedVectors
 
-# Function to download models using requests
+# Function to download files from Google Drive using gdown
 def download_file_from_google_drive(file_id, destination):
-    URL = f"https://drive.google.com/uc?export=download&id={file_id}"
-    session = requests.Session()
-    response = session.get(URL, stream=True)
-    with open(destination, "wb") as f:
-        for chunk in response.iter_content(chunk_size=1024):
-            if chunk:
-                f.write(chunk)
+    gdown.download(f"https://drive.google.com/uc?export=download&id={file_id}", destination, quiet=False)
 
-# Download models from Google Drive
+# Download models from Google Drive using the file IDs
 download_file_from_google_drive('1iN8Hp4cJfEZc2s4vBij7MrKksGRWWJpL', 'word2vec_model.model')
 download_file_from_google_drive('19-vsY_0Gy5EbPOd9qO4mf8rjKvndJfry', 'log_reg_model.pkl')
 download_file_from_google_drive('1XR_jMg2WpZPdzpiB7ZXL-iUGHA0l4S3J', 'nb_model.pkl')
 
-# Load Word2Vec model
+# Load Word2Vec model (make sure it's in binary format)
 word2vec_model = KeyedVectors.load_word2vec_format('word2vec_model.model', binary=True)
 
 # Load Logistic Regression model
@@ -69,4 +63,3 @@ iface3 = gr.Interface(fn=nb_predict,
 iface1.launch()
 iface2.launch()
 iface3.launch()
-