Commit ·
72ea5ef
1
Parent(s): 2670892
Reformatted code explanation
Browse files
README.md
CHANGED
|
@@ -52,13 +52,13 @@ Model performance metrics computed for this model were:
|
|
| 52 |
|
| 53 |
## Install the library
|
| 54 |
|
| 55 |
-
```
|
| 56 |
pip install huggingface_hub
|
| 57 |
```
|
| 58 |
|
| 59 |
## Download the model
|
| 60 |
|
| 61 |
-
```
|
| 62 |
from huggingface_hub import hf_hub_download
|
| 63 |
|
| 64 |
model_path = hf_hub_download(
|
|
@@ -71,7 +71,7 @@ print("Model downloaded to: ", model_path)
|
|
| 71 |
|
| 72 |
## Load the model into PyTorch
|
| 73 |
|
| 74 |
-
```
|
| 75 |
import torch
|
| 76 |
from resnet50 import ResNet50
|
| 77 |
|
|
@@ -83,7 +83,7 @@ model.eval()
|
|
| 83 |
|
| 84 |
## Run inference
|
| 85 |
|
| 86 |
-
```
|
| 87 |
from torchvision import transforms
|
| 88 |
from PIL import Image
|
| 89 |
|
|
|
|
| 52 |
|
| 53 |
## Install the library
|
| 54 |
|
| 55 |
+
```python
|
| 56 |
pip install huggingface_hub
|
| 57 |
```
|
| 58 |
|
| 59 |
## Download the model
|
| 60 |
|
| 61 |
+
```python
|
| 62 |
from huggingface_hub import hf_hub_download
|
| 63 |
|
| 64 |
model_path = hf_hub_download(
|
|
|
|
| 71 |
|
| 72 |
## Load the model into PyTorch
|
| 73 |
|
| 74 |
+
```python
|
| 75 |
import torch
|
| 76 |
from resnet50 import ResNet50
|
| 77 |
|
|
|
|
| 83 |
|
| 84 |
## Run inference
|
| 85 |
|
| 86 |
+
```python
|
| 87 |
from torchvision import transforms
|
| 88 |
from PIL import Image
|
| 89 |
|