Update README.md
Browse files
README.md
CHANGED
|
@@ -71,7 +71,7 @@ inputs = extractor(multi_band_images, return_tensors="pt")
|
|
| 71 |
outputs = model(**inputs)
|
| 72 |
logits = outputs.logits
|
| 73 |
predicted_class = torch.argmax(logits, dim=1)
|
| 74 |
-
|
| 75 |
|
| 76 |
## Training Details
|
| 77 |
|
|
|
|
| 71 |
outputs = model(**inputs)
|
| 72 |
logits = outputs.logits
|
| 73 |
predicted_class = torch.argmax(logits, dim=1)
|
| 74 |
+
```
|
| 75 |
|
| 76 |
## Training Details
|
| 77 |
|