Upload app.py

app.py

@@ -1,3 +1,23 @@
+import gradio as gr
+import joblib
+import pickle
+import numpy as np
+from janome.tokenizer import Tokenizer
+
+model = joblib.load('./doc2vec.pkl')
+with open('document_vecs.pkl', 'rb') as f:
+  document_vecs = pickle.load(f)
+with open('Title.pkl', 'rb') as f:
+  Title = pickle.load(f)
+
+def sep_by_janome(text):
+    t = Tokenizer()
+    tokens = t.tokenize(text)
+    docs=[]
+    for token in tokens:
+        docs.append(token.surface)
+    return docs
+
 def cos_calc(text):
   tokens = sep_by_janome(text)
   input_vec = model.infer_vector(tokens)