Update app.py

app.py

@@ -1,12 +1,16 @@
-import torch
-from transformers import XLNetTokenizer, XLNetForSequenceClassification
+import os
+from transformers import XLNetForSequenceClassification, XLNetTokenizer
 import gradio as gr
 
-# Link to the saved model on Hugging Face Spaces
-model_link = 'https://huggingface.co/spaces/AliArshad/SeverityPrediction/blob/main/severitypredictor.pt'
+# Path where the repository will be cloned
+repo_path = "./severity_prediction_model"
 
-# Load the model directly from the link
-model = XLNetForSequenceClassification.from_pretrained(model_link)
+# Clone the repository containing the model
+os.system(f"git lfs clone https://huggingface.co/spaces/AliArshad/SeverityPrediction {repo_path}")
+
+# Load the model from the cloned repository
+model_path = os.path.join(repo_path, "severitypredictor.pt")
+model = XLNetForSequenceClassification.from_pretrained(model_path)
 
 # Other model setup
 tokenizer = XLNetTokenizer.from_pretrained('xlnet-base-cased')