Update app.py

app.py

@@ -1,14 +1,16 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
-from google_drive_downloader import GoogleDriveDownloader as gdd
+import gdown
+import os
 
 # Set the title of the Streamlit app
 st.title("Text Classification with Hugging Face Transformers")
 
 # Function to download the model from Google Drive
 def download_model_from_drive(file_id, dest_path):
-    gdd.download_file_from_google_drive(file_id=file_id, dest_path=dest_path, unzip=False)
+    url = f'https://drive.google.com/uc?id={file_id}'
+    gdown.download(url, dest_path, quiet=False)
 
 # Download the model files
 with st.spinner("Downloading model..."):