Update app.py

app.py

@@ -7,7 +7,7 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 #Base Model 및 Lora Model 선택
 base_model = "EleutherAI/polyglot-ko-5.8b" # "beomi/KoAlpaca-Polyglot-5.8B"
-lora_weights = "KimSHine/Scenario_Koalpaca_v0_5.8B-lora" # 'KimSHine/Scenario_Koalpaca_5.8B-lora'
+lora_weights = "KimSHine/YEOLLM_5.8B-lora_v3" # 'KimSHine/Scenario_Koalpaca_5.8B-lora'
 load_8bit = True
 
 # Base Model Tokenizer