Update app.py

app.py

@@ -1,14 +1,20 @@
 import streamlit as st
 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, GenerationConfig
+from pydrive2.auth import GoogleAuth
+from pydrive2.drive import GoogleDrive
 
-# Define the URL of your model on Hugging Face Spaces
-model_rpo = 'waqasali1707/Bart-Base-Summarization'
+# Authenticate and create the PyDrive client.
+gauth = GoogleAuth()
+gauth.LocalWebserverAuth()  # Creates a local webserver and automatically handles authentication.
+drive = GoogleDrive(gauth)
+
+# Update this path to your local path where the model is stored
+model_path = '/content/drive/My Drive/bart-base'
 
-# Load the model and tokenizer directly from the URL
 try:
-    tokenizer = AutoTokenizer.from_pretrained(model_rpo)
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_rpo)
+    tokenizer = AutoTokenizer.from_pretrained(model_path)
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
 except Exception as e:
     st.error(f"Failed to load model: {e}")
 
@@ -36,4 +42,4 @@ if text:
             st.write(output)
 
     except Exception as e:
-        st.error(f"An error occurred during summarization: {e}")
+        st.error(f"An error occurred during summarization: {e}")