Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ model, tokenizer = FastLanguageModel.from_pretrained(
|
|
20 |
max_seq_length=max_seq_length,
|
21 |
dtype=dtype,
|
22 |
load_in_4bit=load_in_4bit,
|
23 |
-
token=
|
24 |
)
|
25 |
print("Model and tokenizer loaded successfully.")
|
26 |
|
@@ -140,7 +140,7 @@ model.push_to_hub_merged(
|
|
140 |
"CyberSentinel-initial",
|
141 |
tokenizer,
|
142 |
save_method="merged_16bit",
|
143 |
-
token=
|
144 |
)
|
145 |
print("Model pushed to hub successfully.")
|
146 |
|
|
|
20 |
max_seq_length=max_seq_length,
|
21 |
dtype=dtype,
|
22 |
load_in_4bit=load_in_4bit,
|
23 |
+
token=hf_token
|
24 |
)
|
25 |
print("Model and tokenizer loaded successfully.")
|
26 |
|
|
|
140 |
"CyberSentinel-initial",
|
141 |
tokenizer,
|
142 |
save_method="merged_16bit",
|
143 |
+
token=hf_token
|
144 |
)
|
145 |
print("Model pushed to hub successfully.")
|
146 |
|