jays009 commited on
Commit
1db589c
·
verified ·
1 Parent(s): f6d8ad8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -6
README.md CHANGED
@@ -1,12 +1,15 @@
1
  ---
2
- model_name: "Wheat Anomaly Detection Model"
3
  tags:
4
- - pytorch
5
- - resnet
6
- - agriculture
7
- - anomaly-detection
8
  license: apache-2.0
9
  library_name: pytorch
 
 
 
10
  ---
11
 
12
  # Wheat Anomaly Detection Model
@@ -37,4 +40,4 @@ inputs = inputs.to(device)
37
  with torch.no_grad():
38
  outputs = model(inputs)
39
  predicted_class = torch.argmax(outputs.logits, dim=1)
40
- print(f"Predicted Class: {predicted_class.item()}")
 
1
  ---
2
+ model_name: Wheat Anomaly Detection Model
3
  tags:
4
+ - pytorch
5
+ - resnet
6
+ - agriculture
7
+ - anomaly-detection
8
  license: apache-2.0
9
  library_name: pytorch
10
+ pipeline_tag: image-classification
11
+ language:
12
+ - en
13
  ---
14
 
15
  # Wheat Anomaly Detection Model
 
40
  with torch.no_grad():
41
  outputs = model(inputs)
42
  predicted_class = torch.argmax(outputs.logits, dim=1)
43
+ print(f"Predicted Class: {predicted_class.item()}")