Update README.md
Browse files
README.md
CHANGED
|
@@ -69,7 +69,7 @@ from PIL import Image
|
|
| 69 |
import torch
|
| 70 |
|
| 71 |
# Load model and processor
|
| 72 |
-
model_name = "prithivMLmods/
|
| 73 |
model = SiglipForImageClassification.from_pretrained(model_name)
|
| 74 |
processor = AutoImageProcessor.from_pretrained(model_name)
|
| 75 |
|
|
|
|
| 69 |
import torch
|
| 70 |
|
| 71 |
# Load model and processor
|
| 72 |
+
model_name = "prithivMLmods/Forest-Fire-Detection" # Update with actual model name on Hugging Face
|
| 73 |
model = SiglipForImageClassification.from_pretrained(model_name)
|
| 74 |
processor = AutoImageProcessor.from_pretrained(model_name)
|
| 75 |
|