infinityy commited on
Commit
d86218d
·
verified ·
1 Parent(s): 3dd8d72

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ supabase: Client = create_client(supabase_url, supabase_anon_key)
20
 
21
  # Load BERT model & tokenizer
22
  tokenizer = AutoTokenizer.from_pretrained("distilbert-base-cased")
23
- model = AutoModelForSequenceClassification.from_pretrained('distilbert-base-cased', num_labels=5,output_hidden_states=True,trust_remote_code=True)
24
  from safetensors.torch import load_file
25
  state_dict = load_file("model (4).safetensors")
26
 
 
20
 
21
  # Load BERT model & tokenizer
22
  tokenizer = AutoTokenizer.from_pretrained("distilbert-base-cased")
23
+ model = AutoModelForSequenceClassification.from_pretrained('distilbert-base-cased', num_labels=5,output_hidden_states=True)
24
  from safetensors.torch import load_file
25
  state_dict = load_file("model (4).safetensors")
26