Spaces:

Cyplus72
/

nsmc-sentiment-lora

Sleeping

Cyplus72 commited on Nov 5

Commit

3455393

verified ·

1 Parent(s): bebcad5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ from peft import PeftModel
 print("모델 로드 중...")
 BASE_MODEL = "klue/bert-base"
-LORA_MODEL = "None" # 여러분의 Model
 tokenizer = AutoTokenizer.from_pretrained(LORA_MODEL)
 base_model = AutoModelForSequenceClassification.from_pretrained(

 print("모델 로드 중...")
 BASE_MODEL = "klue/bert-base"
+LORA_MODEL = "Cyplus72/nsmc-sentiment-lora" # 여러분의 Model
 tokenizer = AutoTokenizer.from_pretrained(LORA_MODEL)
 base_model = AutoModelForSequenceClassification.from_pretrained(