AliArshad commited on
Commit
281cbf1
1 Parent(s): cd7b9e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
3
  import gradio as gr
4
 
5
  # Path to the saved model in Hugging Face Spaces
6
- model_name = 'AliArshad/SeverityPrediction' # Replace with your actual username and model name
7
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
8
  tokenizer = AutoTokenizer.from_pretrained(model_name)
9
 
 
3
  import gradio as gr
4
 
5
  # Path to the saved model in Hugging Face Spaces
6
+ model_name = 'AliArshad/SeverityPredictor' # Replace with your actual username and model name
7
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
8
  tokenizer = AutoTokenizer.from_pretrained(model_name)
9