Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,13 +12,13 @@ import streamlit as st
|
|
12 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
13 |
|
14 |
# Load the base model
|
15 |
-
base_model_name = "
|
16 |
tokenizer = AutoTokenizer.from_pretrained(base_model_name)
|
17 |
model = AutoModelForSequenceClassification.from_pretrained(base_model_name)
|
18 |
|
19 |
# Load the adapter configuration and model files
|
20 |
-
adapter_config_path = "config.json"
|
21 |
-
adapter_model_path = "model.safetensors"
|
22 |
|
23 |
# Load the adapter into the model
|
24 |
adapter_name = "custom_adapter" # Define your adapter name
|
|
|
12 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
13 |
|
14 |
# Load the base model
|
15 |
+
base_model_name = "bert-base-uncased"
|
16 |
tokenizer = AutoTokenizer.from_pretrained(base_model_name)
|
17 |
model = AutoModelForSequenceClassification.from_pretrained(base_model_name)
|
18 |
|
19 |
# Load the adapter configuration and model files
|
20 |
+
adapter_config_path = "./config.json"
|
21 |
+
adapter_model_path = "./model.safetensors"
|
22 |
|
23 |
# Load the adapter into the model
|
24 |
adapter_name = "custom_adapter" # Define your adapter name
|