epochs test

app.py

@@ -5,6 +5,10 @@ import torch.nn as nn
 import torch.optim as optim
 from torchvision import datasets, transforms
 from torch.utils.data import DataLoader
+from huggingface_hub import notebook_login
+from huggingface_hub import Repository
+
+notebook_login()
 
 class SimpleNet(nn.Module):
     def __init__(self):
@@ -45,6 +49,10 @@ def train_model(epochs):
     
     # Save the model checkpoint
     torch.save(model.state_dict(), "simple_net.pth")
+    repo = Repository(local_dir=" https://huggingface.co/spaces/Transduce/vc_demo_clean", repo_id="Transduce/testing_cuda", use_auth_token=True)
+    repo.git_add(file_paths="simple_net.pth")
+    repo.git_commit("model checkpoint")
+    repo.git_push()
     return "Training completed and model saved."
 
 # Define the Gradio interface