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 |
|