rishabh914 commited on
Commit
f7d385f
·
verified ·
1 Parent(s): 3de4040

Add model documentation

Browse files
Files changed (1) hide show
  1. README.md +36 -3
README.md CHANGED
@@ -1,3 +1,36 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - image-classification
5
+ - plant-disease
6
+ - tensorflow
7
+ - keras
8
+ library_name: tensorflow
9
+ ---
10
+
11
+ # Leaf Disease Detection Model
12
+
13
+ This model can detect diseases in plant leaves using EfficientNetB3 architecture.
14
+
15
+ ## Usage
16
+
17
+ ```python
18
+ from huggingface_hub import hf_hub_download
19
+ import tensorflow as tf
20
+ import json
21
+
22
+ # Download model and class indices
23
+ model_path = hf_hub_download(repo_id="Rvish-glitch/leaf-disease-detection", filename="final_model.h5")
24
+ indices_path = hf_hub_download(repo_id="Rvish-glitch/leaf-disease-detection", filename="class_indices.json")
25
+
26
+ # Load model
27
+ model = tf.keras.models.load_model(model_path)
28
+
29
+ # Load class indices
30
+ with open(indices_path, 'r') as f:
31
+ class_indices = json.load(f)
32
+ ```
33
+
34
+ ## Classes
35
+
36
+ The model can detect various plant diseases across different crops including tomato, potato, apple, and corn.