Update app.py

app.py

@@ -1,16 +1,22 @@
 import torch
+import requests
 from transformers import XLNetTokenizer, XLNetForSequenceClassification
 import gradio as gr
 
 # Link to the saved model on Hugging Face Spaces
-model_link = 'https://huggingface.co/spaces/AliArshad/SeverityPrediction/resolve/main/severitypredictor.pt'
+model_link = 'https://huggingface.co/spaces/AliArshad/SeverityPrediction/blob/main/severitypredictor.pt'
 
-# Load the model directly from the link
-model = XLNetForSequenceClassification.from_pretrained(model_link, from_tf=False)
+# Download the model file
+response = requests.get(model_link)
+model_path = 'severitypredictor.pt'
+with open(model_path, 'wb') as f:
+    f.write(response.content)
+
+# Load the downloaded model using PyTorch
+model = torch.load(model_path)
 
 # Other model setup
 tokenizer = XLNetTokenizer.from_pretrained('xlnet-base-cased')
-model.eval()
 
 # Function for prediction
 def xl_net_predict(text):