Update README.md

README.md

@@ -24,18 +24,11 @@ This model is a fine-tuned version of [roberta-base](https://huggingface.co/robe
 # Usage
 
 ```python
-test_query= [
-    "The 2022 FIFA World Cup was the 22nd FIFA World Cup, the quadrennial world championship for national football teams organized by FIFA.",
-    "Argentina were crowned the champions after winning the final against the title holder France 4-2 on penalties following a 3-3 draw after extra time.", 
-    "It was Argentina's third title and their first since 1986, as well being the first nation from outside of Europe to win the tournament since 2002.", 
-    "French player Kylian Mbappé became the first player to score a hat-trick in a World Cup final since Geoff Hurst in the 1966 final and won the Golden Boot as he scored the most goals (eight) during the tournament.", 
-    "Argentine captain Lionel Messi was voted the tournament's best player, winning the Golden Ball. Teammates Emiliano Martínez and Enzo Fernández won the Golden Glove, awarded to the tournament's best goalkeeper, and the Young Player Award, awarded to the tournament's best young player."
-]
-entities_per_query = model.predict(test_query)
-
-for entities in entities_per_query:
-    for entity in entities:
-        print(entity["span"], "-->", entity["label"])
+  from span_marker import SpanMarkerModel
+  
+  model = SpanMarkerModel.from_pretrained("krinal/span-marker-robert-base")
+  
+  ner_result = model.predict("Argentine captain Lionel Messi won Golden Ball at FIFA world cup 2022")
 ```
 
 ## Training and evaluation data