Update app.py

app.py

@@ -17,13 +17,15 @@ current_num = os.getenv("NUM")
 print(f"stage ${current_num}")
 
 api = HfApi(token=hf_token)
-models = f"dad1909/CyberSentinel-{current_num}"
+# models = f"dad1909/CyberSentinel-{current_num}"
+
+model_base = "dad1909/cybersentinal-2.0"
 
 print("Starting model and tokenizer loading...")
 
 # Load the model and tokenizer
 model, tokenizer = FastLanguageModel.from_pretrained(
-    model_name=models,
+    model_name=model_base,
     max_seq_length=max_seq_length,
     dtype=dtype,
     load_in_4bit=load_in_4bit,
@@ -141,7 +143,7 @@ print("Training completed.")
 num = int(current_num)
 num += 1
 
-uploads_models = f"CyberSentinel-{str(num)}"
+uploads_models = f"cybersentinal-2.0-{str(num)}"
 
 print("Saving the trained model...")
 model.save_pretrained_merged("model", tokenizer, save_method="merged_16bit")