Upload folder using huggingface_hub

.idea/workspace.xml

@@ -54,7 +54,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1760056680813</updated>
-      <workItem from="1760056681869" duration="4675000" />
+      <workItem from="1760056681869" duration="5346000" />
     </task>
     <servers />
   </component>

app.py

@@ -43,11 +43,8 @@ _ = snapshot_download(
     local_dir_use_symlinks=False,
     revision="main",
 )
-checkpoint_path = "checkpoints/Puffin-Base.pth"
-checkpoint = torch.load(checkpoint_path, map_location='cpu')
-_ = model.load_state_dict(checkpoint, strict=False)
-
-os.remove(checkpoint)
+_ = model.load_state_dict(torch.load("checkpoints/Puffin-Base.pth", map_location='cpu'), strict=False)
+os.remove("checkpoints/Puffin-Base.pth")
 
 _ = snapshot_download(
     repo_id="wusize/Puffin",