blizet commited on
Commit
0bf6b22
·
verified ·
1 Parent(s): 7f1f0a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from transformers import (
9
  from peft import LoraConfig
10
 
11
  # Define the model name and fine-tuned model output path
12
- model_name = "llm_pretrained_merged"
13
 
14
  ################################################################################
15
  # LoRA Configuration Parameters
 
9
  from peft import LoraConfig
10
 
11
  # Define the model name and fine-tuned model output path
12
+ model_name = "blizet/Llama-Phishing-Finetune"
13
 
14
  ################################################################################
15
  # LoRA Configuration Parameters