Update README.md

README.md

@@ -1,22 +1,27 @@
 ---
 language: en
-license: mit
+license: apache-2.0
 tags:
-  - metaphor-detection
-  - bert
-  - text-classification
-  - nlp
-  - transformer
+- metaphor-detection
+- bert
+- text-classification
+- nlp
+- transformer
 model-index:
-  - name: Fine-Tuned Metaphor Detection Model
-    results:
-      - task:
-          name: text-classification
-          type: text-classification  # This line was missing
-        metrics:
-          - name: Accuracy
-            value: 72
-            type: accuracy
+- name: Fine-Tuned Metaphor Detection Model
+  results:
+  - task:
+      name: text-classification
+      type: text-classification
+    metrics:
+    - name: Accuracy
+      value: 72
+      type: accuracy
+metrics:
+- accuracy
+base_model:
+- google-bert/bert-base-uncased
+pipeline_tag: text-classification
 ---
 
 # Fine-Tuned Metaphor Detection Model
@@ -55,4 +60,4 @@ with torch.no_grad():
     logits = outputs.logits
     prediction = torch.argmax(logits, dim=-1)
 
-print("Prediction:", "Metaphor" if prediction.item() == 1 else "Literal")
+print("Prediction:", "Metaphor" if prediction.item() == 1 else "Literal")