Update app.py

app.py

@@ -13,11 +13,14 @@ dtype = None
 load_in_4bit = True
 hf_token = os.getenv("HF_TOKEN")
 
+pre_models = "dad1909/CyberSentinel-3"
+uploads_models = "CyberSentinel-4"
+
 print("Starting model and tokenizer loading...")
 
 # Load the model and tokenizer
 model, tokenizer = FastLanguageModel.from_pretrained(
-    model_name="dad1909/CyberSentinel-2",
+    model_name=pre_models,
     max_seq_length=max_seq_length,
     dtype=dtype,
     load_in_4bit=load_in_4bit,
@@ -138,7 +141,7 @@ print("Model saved successfully.")
 
 print("Pushing the model to the hub...")
 model.push_to_hub_merged(
-    "CyberSentinel-3",
+    uploads_models,
     tokenizer,
     save_method="merged_16bit",
     token=hf_token