create model card

README.md

@@ -0,0 +1,31 @@
+How to use:
+"""
+from collections import deque
+from  bs4 import BeautifulSoup
+import requests
+
+def dialog(context):
+  
+    keyword = generate('keyword: ' + ' '.join(context), num_beams=2,)[0]
+    knowlege = ''
+    if keyword != 'no_keywords':
+        resp = requests.get(f"https://en.wikipedia.org/wiki/{keyword}")
+        root = BeautifulSoup(resp.content, "html.parser")
+        knowlege ="knowlege: " + " ".join([_.text.strip() for _ in root.find("div", class_="mw-body-content mw-content-ltr").find_all("p", limit=2)])
+
+    answ = generate(f'dialog: ' + knowlege + ' '.join(context), num_beams=3,
+                    do_sample=True, temperature=1.1, encoder_no_repeat_ngram_size=5, 
+                    no_repeat_ngram_size=5, 
+                    max_new_tokens  = 30)[0]
+    return answ
+
+context =deque([], maxlen=4)
+while True:
+    text = input()
+    text = 'user1>>: ' + text
+    context.append(text)
+    answ = dialog(context)
+    context.append('user2>>: ' + answ)
+
+    print('bot: ', answ)
+"""