ddiddu commited on
Commit
44e7e38
·
1 Parent(s): 4051bd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- # gr.outputs.Textbox(label='Similarity')
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'''],