hii

app.py

@@ -5,10 +5,9 @@ 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 HfFolder
-from huggingface_hub import Repository
-HfFolder.save_token("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")
+# from huggingface_hub import HfFolder
+# from huggingface_hub import Repository
+# HfFolder.save_token("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")
 
 class SimpleNet(nn.Module):
     def __init__(self):
@@ -49,10 +48,6 @@ 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.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