Update app.py

app.py

@@ -1,26 +1,15 @@
 import torch
 from transformers import XLNetTokenizer, XLNetForSequenceClassification
 import gradio as gr
-from pydrive.auth import GoogleAuth
-from pydrive.drive import GoogleDrive
 
-# Authenticate and create GoogleDrive instance
-gauth = GoogleAuth()
-gauth.LocalWebserverAuth()
-drive = GoogleDrive(gauth)
+# Link to the saved model on Hugging Face Spaces
+model_link = 'https://huggingface.co/spaces/AliArshad/SeverityPrediction/blob/main/severitypredictor.pt'
 
-# ID of the file in Google Drive
-file_id = '1-7O5gAFgcIzgJ68WkSSpmh1H6kJL6fAO'  # Replace this with your file's ID from Google Drive
-destination_path = '/content/XLNet_model_project_Core.pt'  # Path to save the downloaded model file
+# Load the model directly from the link
+model = XLNetForSequenceClassification.from_pretrained(model_link)
 
-# Download the model file from Google Drive
-downloaded_file = drive.CreateFile({'id': file_id})
-downloaded_file.GetContentFile(destination_path)
-
-# Load the saved model
+# Other model setup
 tokenizer = XLNetTokenizer.from_pretrained('xlnet-base-cased')
-model = XLNetForSequenceClassification.from_pretrained('xlnet-base-cased', num_labels=2)
-model.load_state_dict(torch.load(destination_path))
 model.eval()
 
 # Function for prediction