Update app.py

app.py

@@ -3,11 +3,10 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import gradio as gr
 
 # Path to the saved model in Hugging Face Spaces
-model_path = 'https://huggingface.co/spaces/AliArshad/SeverityPrediction/blob/main/XLNet_model_project_Core.pt'
+model_name = 'AliArshad/SeverityPrediction'  # Replace with your actual username and model name
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-# Load the model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained('xlnet-base-cased')
-model = AutoModelForSequenceClassification.from_pretrained(model_path)
 
 # Function for prediction
 def xl_net_predict(text):