Update app.py
Browse files
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
|
| 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 |
|