surmount01 commited on
Commit
76c5393
·
verified ·
1 Parent(s): 5127e96

Rename app.py.txt to app.py

Browse files
Files changed (1) hide show
  1. app.py.txt → app.py +1 -1
app.py.txt → app.py RENAMED
@@ -11,7 +11,7 @@ from peft import PeftModel
11
  print("모델 로드 중...")
12
 
13
  BASE_MODEL = "klue/bert-base"
14
- LORA_MODEL = "surmount01" # 여러분의 Model
15
 
16
  tokenizer = AutoTokenizer.from_pretrained(LORA_MODEL)
17
  base_model = AutoModelForSequenceClassification.from_pretrained(
 
11
  print("모델 로드 중...")
12
 
13
  BASE_MODEL = "klue/bert-base"
14
+ LORA_MODEL = "surmount01/nsmc-sentiment-lora" # 여러분의 Model
15
 
16
  tokenizer = AutoTokenizer.from_pretrained(LORA_MODEL)
17
  base_model = AutoModelForSequenceClassification.from_pretrained(