Update train_model.py

train_model.py

@@ -139,7 +139,7 @@ def main():
     tokenizer.save_pretrained(output_dir)
     
     # Push to Hugging Face Hub
-    model_repo = f"your-username/{args.model_name}"  # Replace 'your-username' with your actual username
+    model_repo = f"Vishwas1/{args.model_name}"  # Replace 'your-username' with your actual username
     try:
         api.create_repo(repo_id=model_repo, private=False, token=hf_token)
     except Exception as e: