Update README.md
Browse files
README.md
CHANGED
|
@@ -7,14 +7,4 @@ tags:
|
|
| 7 |
- construction
|
| 8 |
- drawings
|
| 9 |
- PDF
|
| 10 |
-
---
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
from huggingface_hub import hf_hub_download
|
| 14 |
-
import torch
|
| 15 |
-
|
| 16 |
-
# Download the model weights
|
| 17 |
-
model_path = hf_hub_download(repo_id="your-username/pdf-drawing-classifier", filename="drawing_classifier.pth")
|
| 18 |
-
|
| 19 |
-
# Load it into your ResNet architecture
|
| 20 |
-
model.load_state_dict(torch.load(model_path, map_location="cpu", weights_only=True))
|
|
|
|
| 7 |
- construction
|
| 8 |
- drawings
|
| 9 |
- PDF
|
| 10 |
+
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|