Hugging Face's logo

Spaces:
KenLumod
/
ML-Project
Sleeping

App Files Community
KenLumod commited on
Commit
8242585
·
verified ·
1 Parent(s): fa6e8e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ model_id = "KenLumod/ML-Fake-Real-News-Detector-Final"
5
 
6
  # Force reload model with updated config
7
  config = AutoConfig.from_pretrained(model_id)
8
- config.id2label = {0: "Fake News", 1: "Real News"} # Force override
9
  config.label2id = {v: k for k, v in config.id2label.items()}
10
 
11
  # Load the model using AutoModelForSequenceClassification
 
5
 
6
  # Force reload model with updated config
7
  config = AutoConfig.from_pretrained(model_id)
8
+ config.id2label = {1: "Fake News", 0: "Real News"} # Force override
9
  config.label2id = {v: k for k, v in config.id2label.items()}
10
 
11
  # Load the model using AutoModelForSequenceClassification