Preetham04 commited on
Commit
08cc45d
1 Parent(s): be2ac5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 = "./model.safetensors"
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