Update app.py

app.py

@@ -1,16 +1,12 @@
-import os
-from transformers import XLNetForSequenceClassification, XLNetTokenizer
+import torch
+from transformers import XLNetTokenizer, XLNetForSequenceClassification
 import gradio as gr
 
-# Path where the repository will be cloned
-repo_path = "./severity_prediction_model"
+# Link to the saved model on Hugging Face Spaces
+model_link = 'https://huggingface.co/spaces/AliArshad/SeverityPrediction/resolve/main/severitypredictor.pt'
 
-# 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)
+# Load the model directly from the link
+model = XLNetForSequenceClassification.from_pretrained(model_link, from_tf=False)
 
 # Other model setup
 tokenizer = XLNetTokenizer.from_pretrained('xlnet-base-cased')