roncc13 commited on
Commit
e7b18a3
·
verified ·
1 Parent(s): 547c955

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from theme import custom_css, header, card_start, card_end
7
  # --------------------------
8
  # Model setup
9
  # --------------------------
10
- MODEL_ID = "roncc13/trainCMDBERT-sample"
11
 
12
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
13
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_ID)
 
7
  # --------------------------
8
  # Model setup
9
  # --------------------------
10
+ MODEL_ID = "roncc13/autotrain-ixzm9-t6dbc"
11
 
12
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
13
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_ID)