Abuzaid01 commited on
Commit
fef7539
Β·
verified Β·
1 Parent(s): 8f63c21

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -15
README.md CHANGED
@@ -1,17 +1,4 @@
1
- readme_text = f"""
2
- ---
3
- language: en
4
- library_name: pytorch
5
- tags:
6
- - image-classification
7
- - efficientnet
8
- - fruits
9
- - vegetables
10
- datasets:
11
- - kritikseth/fruit-and-vegetable-image-recognition
12
- license: mit
13
- pipeline_tag: image-classification
14
- ---
15
 
16
  # EfficientNet-B0 Fruit & Vegetable Classifier 🍎πŸ₯•πŸŒ½
17
 
@@ -50,4 +37,4 @@ model.classifier = torch.nn.Sequential(
50
 
51
  # Load weights
52
  model.load_state_dict(torch.load("pytorch_model.bin", map_location="cpu"))
53
- model.eval()
 
1
+
 
 
 
 
 
 
 
 
 
 
 
 
 
2
 
3
  # EfficientNet-B0 Fruit & Vegetable Classifier 🍎πŸ₯•πŸŒ½
4
 
 
37
 
38
  # Load weights
39
  model.load_state_dict(torch.load("pytorch_model.bin", map_location="cpu"))
40
+ model.eval()