breadlicker45 commited on
Commit
707a523
·
verified ·
1 Parent(s): 8148bc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,8 +17,8 @@ print(f"Loading model: {MODEL_ID}...")
17
  try:
18
  # Tokenizer can still be loaded automatically
19
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, token=HF_TOKEN, trust_remote_code=True)
20
- commit_hash = "4da4ceaf8f9b6fbba3528d38dba526f39ac75970"
21
- model = AutoModelForSequenceClassification.from_pretrained(MODEL_ID, token=HF_TOKEN, trust_remote_code=True, revision=commit_hash)
22
 
23
  # Move the model to the selected device
24
  model.to(device)
 
17
  try:
18
  # Tokenizer can still be loaded automatically
19
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, token=HF_TOKEN, trust_remote_code=True)
20
+
21
+ model = AutoModelForSequenceClassification.from_pretrained(MODEL_ID, token=HF_TOKEN, trust_remote_code=True)
22
 
23
  # Move the model to the selected device
24
  model.to(device)