Update app.py
Browse files
app.py
CHANGED
|
@@ -17,7 +17,7 @@ print("Starting model and tokenizer loading...")
|
|
| 17 |
|
| 18 |
# Load the model and tokenizer
|
| 19 |
model, tokenizer = FastLanguageModel.from_pretrained(
|
| 20 |
-
model_name="dad1909/CyberSentinel-
|
| 21 |
max_seq_length=max_seq_length,
|
| 22 |
dtype=dtype,
|
| 23 |
load_in_4bit=load_in_4bit,
|
|
@@ -138,7 +138,7 @@ print("Model saved successfully.")
|
|
| 138 |
|
| 139 |
print("Pushing the model to the hub...")
|
| 140 |
model.push_to_hub_merged(
|
| 141 |
-
"CyberSentinel-
|
| 142 |
tokenizer,
|
| 143 |
save_method="merged_16bit",
|
| 144 |
token=hf_token
|
|
|
|
| 17 |
|
| 18 |
# Load the model and tokenizer
|
| 19 |
model, tokenizer = FastLanguageModel.from_pretrained(
|
| 20 |
+
model_name="dad1909/CyberSentinel-2",
|
| 21 |
max_seq_length=max_seq_length,
|
| 22 |
dtype=dtype,
|
| 23 |
load_in_4bit=load_in_4bit,
|
|
|
|
| 138 |
|
| 139 |
print("Pushing the model to the hub...")
|
| 140 |
model.push_to_hub_merged(
|
| 141 |
+
"CyberSentinel-3",
|
| 142 |
tokenizer,
|
| 143 |
save_method="merged_16bit",
|
| 144 |
token=hf_token
|