Update app.py

app.py

@@ -1,20 +1,14 @@
 import streamlit as st
 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, GenerationConfig
-from pydrive2.auth import GoogleAuth
-from pydrive2.drive import GoogleDrive
 
-# 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'
+# Define the URL of your model on Hugging Face Spaces
+model_url = 'https://huggingface.co/your-username/your-model-name/resolve/main/'
 
+# Load the model and tokenizer directly from the URL
 try:
-    tokenizer = AutoTokenizer.from_pretrained(model_path)
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
+    tokenizer = AutoTokenizer.from_pretrained(model_url)
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_url)
 except Exception as e:
     st.error(f"Failed to load model: {e}")
 
@@ -42,4 +36,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}")