Update app.py

app.py

@@ -23,12 +23,12 @@ tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
 #base_model = AutoModelForCausalLM.from_pretrained("EleutherAI/polyglot-ko-3.8b")
 #tokenizer = AutoTokenizer.from_pretrained("EleutherAI/polyglot-ko-3.8b")
 base_model.eval()
-#base_model.config.use_cache = True
+base_model.config.use_cache = True
 
 
 model = PeftModel.from_pretrained(base_model, peft_model_id, device_map="auto")
 model.eval()
-#model.config.use_cache = True
+model.config.use_cache = True
 
 
 mbti_bert_model_name = "Lanvizu/fine-tuned-klue-bert-base_model_11"