Update app.py
Browse files
app.py
CHANGED
|
@@ -200,7 +200,7 @@ import gradio as gr
|
|
| 200 |
def greet(paper):
|
| 201 |
title_tfidf, categories_tfidf, abstract_tfidf, similarity_tfidf = get_recommendations_TFIDF(paper)
|
| 202 |
title_doc2vec, categories_doc2vec, abstract_doc2vec, similarity_doc2vec = get_recommendations_Doc2Vec(paper)
|
| 203 |
-
return title_tfidf, categories_tfidf, abstract_tfidf, title_doc2vec, categories_doc2vec, abstract_doc2vec
|
| 204 |
|
| 205 |
title = '''SimSearch:\n
|
| 206 |
A Similarity Search Tool for Research Paper Abstracts using NLP'''
|
|
@@ -212,10 +212,11 @@ demo = gr.Interface(
|
|
| 212 |
outputs=[gr.outputs.Textbox(label='''TFIDF Title'''),
|
| 213 |
gr.outputs.Textbox(label='TFIDF Categories'),
|
| 214 |
gr.outputs.Textbox(label='TFIDF Abstract'),
|
| 215 |
-
|
| 216 |
gr.outputs.Textbox(label='''Doc2Vec Title'''),
|
| 217 |
gr.outputs.Textbox(label='Doc2Vec Categories'),
|
| 218 |
gr.outputs.Textbox(label='Doc2Vec Abstract'),
|
|
|
|
| 219 |
],
|
| 220 |
examples = [
|
| 221 |
# ['''In this paper we consider permutations of sequences of partitions, obtaining\na result which parallels von Neumann's theorem on permutations of dense\nsequences and uniformly distributed sequences of points.\n'''],
|
|
|
|
| 200 |
def greet(paper):
|
| 201 |
title_tfidf, categories_tfidf, abstract_tfidf, similarity_tfidf = get_recommendations_TFIDF(paper)
|
| 202 |
title_doc2vec, categories_doc2vec, abstract_doc2vec, similarity_doc2vec = get_recommendations_Doc2Vec(paper)
|
| 203 |
+
return title_tfidf, categories_tfidf, abstract_tfidf, similarity_tfidf, title_doc2vec, categories_doc2vec, abstract_doc2vec, similarity_doc2vec
|
| 204 |
|
| 205 |
title = '''SimSearch:\n
|
| 206 |
A Similarity Search Tool for Research Paper Abstracts using NLP'''
|
|
|
|
| 212 |
outputs=[gr.outputs.Textbox(label='''TFIDF Title'''),
|
| 213 |
gr.outputs.Textbox(label='TFIDF Categories'),
|
| 214 |
gr.outputs.Textbox(label='TFIDF Abstract'),
|
| 215 |
+
gr.outputs.Textbox(label='TFIDF Similarity')
|
| 216 |
gr.outputs.Textbox(label='''Doc2Vec Title'''),
|
| 217 |
gr.outputs.Textbox(label='Doc2Vec Categories'),
|
| 218 |
gr.outputs.Textbox(label='Doc2Vec Abstract'),
|
| 219 |
+
gr.outputs.Textbox(label='Doc2Vec Similarity'),
|
| 220 |
],
|
| 221 |
examples = [
|
| 222 |
# ['''In this paper we consider permutations of sequences of partitions, obtaining\na result which parallels von Neumann's theorem on permutations of dense\nsequences and uniformly distributed sequences of points.\n'''],
|