Update README.md

README.md

@@ -36,7 +36,7 @@ Sample
 {
     "Question": "who is the tallest host?",
     "Answer": ["sam bee","sam bee"],
-    "Tweet": "Don't believe @ConanOBrien's height lies. Sam Bee is the tallest host in late night. #alternativefacts\\u2014 Full Frontal (@FullFrontalSamB) January 22, 2017",
+    "Tweet": "Don't believe @ConanOBrien's height lies. Sam Bee is the tallest host in late night. #alternativefacts\\\\u2014 Full Frontal (@FullFrontalSamB) January 22, 2017",
     "qid": "3554ee17d86b678be34c4dc2c04e334f"
 }
 ```
@@ -75,9 +75,12 @@ def get_answer(question, context):
     attention_mask = inputs.attention_mask.to('cuda')
     output = model.generate(input_ids, attention_mask=attention_mask)
     return tokenizer.decode(output[0], skip_special_tokens=True)
+
+
+context = "Don't believe @ConanOBrien's height lies. Sam Bee is the tallest host in late night. #alternativefacts\\\\u2014 Full Frontal (@FullFrontalSamB) January 22, 2017"
+question = "who is the tallest host?"          
     
-    
-get_answer('here goes your question', 'And here the context/tweet...')
+get_answer(question, context)
 ```
 
 Created by: [Narrativa](https://www.narrativa.com/)