Update README.md

README.md

@@ -1,12 +1,15 @@
 ---
-model_name: "Wheat Anomaly Detection Model"
+model_name: Wheat Anomaly Detection Model
 tags:
-  - pytorch
-  - resnet
-  - agriculture
-  - anomaly-detection
+- pytorch
+- resnet
+- agriculture
+- anomaly-detection
 license: apache-2.0
 library_name: pytorch
+pipeline_tag: image-classification
+language:
+- en
 ---
 
 # Wheat Anomaly Detection Model
@@ -37,4 +40,4 @@ inputs = inputs.to(device)
 with torch.no_grad():
     outputs = model(inputs)
     predicted_class = torch.argmax(outputs.logits, dim=1)
-    print(f"Predicted Class: {predicted_class.item()}")
+    print(f"Predicted Class: {predicted_class.item()}")