update

upload_to_hf.py

@@ -32,8 +32,6 @@ PhaseNet-TF is an advanced deep learning model for automatic seismic phase picki
 - **Spectrogram Size**: 64 × 4800 (frequency × time)
 - **Input Channels**: 6 (3 real + 3 imaginary spectrogram channels)
 - **Output Classes**: 4 (noise, P, S, PS)
-
-## Load the checkpoint
 """
 
 def create_config_json(model_path, version=None):
@@ -96,20 +94,6 @@ This repository contains two versions of the PhaseNet-TF Alaska model:
 - **Config**: `config_iter2.json`
 - **Documentation**: [README_iter2.md](README_iter2.md)
 
-## Quick Start
-
-### Load Iteration 1
-```python
-import torch
-checkpoint = torch.load("pytorch_model_iter1.bin", map_location="cpu")
-```
-
-### Load Iteration 2
-```python
-import torch
-checkpoint = torch.load("pytorch_model_iter2.bin", map_location="cpu")
-```
-
 ## Model Architecture
 
 - **Backbone**: DeepLabV3Plus with ResNet34 encoder
@@ -226,10 +210,6 @@ def upload_model_to_hf(
     
     # Create and upload README.md
     model_card = create_model_card(version)
-    if version:
-        model_card += f'checkpoint = torch.load("pytorch_model_{version}.bin", map_location="cpu")'
-    else:
-        model_card += 'checkpoint = torch.load("pytorch_model.bin", map_location="cpu")'
     
     model_card += """