Update app.py

app.py

@@ -203,6 +203,35 @@ if (runModel=='1'):
     tokenizer_files = os.listdir(tokenizer_path)
     tokenizer_files = [file for file in tokenizer_files]
     print("Specific files in tokenizer directory:", tokenizer_files)
+
+    #for push repository
+    repo_name = "Reyad-Ahmmed/hf-data-timeframe" 
+
+    # Your repository name 
+    api_token = os.getenv("hf-token")  # Retrieve the API token from environment variable
+ 
+    if not api_token:
+        raise ValueError("API token not found. Please set the HF_API_TOKEN environment variable.")
+
+    # Create repository (if not already created)
+    api = HfApi()
+    create_repo(repo_id=repo_name, token=api_token, exist_ok=True)
+
+    # Upload the model and tokenizer to the Hugging Face repository
+    upload_folder(
+        folder_path=model_path,
+        path_in_repo="data-timeframe_model",
+        repo_id=repo_name,
+        token=api_token,
+        commit_message="Update fine-tuned model"
+    )
+    upload_folder(
+        folder_path=tokenizer_path,
+        path_in_repo="data-timeframe_tokenizer",
+        repo_id=repo_name,
+        token=api_token,
+        commit_message="Update fine-tuned tokenizer"
+    )
     
 else:
     print('Load Pre-trained')