Update app.py

app.py

@@ -1,12 +1,22 @@
 import gradio as gr
 import joblib
-import gdown
+import requests
 from gensim.models import KeyedVectors
 
-# Download models from Google Drive using their file IDs
-gdown.download("https://drive.google.com/uc?export=download&id=1iN8Hp4cJfEZc2s4vBij7MrKksGRWWJpL", "word2vec_model.model", quiet=False)
-gdown.download("https://drive.google.com/uc?export=download&id=19-vsY_0Gy5EbPOd9qO4mf8rjKvndJfry", "log_reg_model.pkl", quiet=False)
-gdown.download("https://drive.google.com/uc?export=download&id=1XR_jMg2WpZPdzpiB7ZXL-iUGHA0l4S3J", "nb_model.pkl", quiet=False)
+# Function to download models using requests
+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)
+
+# Download models from Google Drive
+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
 word2vec_model = KeyedVectors.load_word2vec_format('word2vec_model.model', binary=True)
@@ -59,3 +69,4 @@ iface3 = gr.Interface(fn=nb_predict,
 iface1.launch()
 iface2.launch()
 iface3.launch()
+