pritamdeka commited on
Commit
e7b9e7c
Β·
1 Parent(s): b9f3350

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -264,8 +264,8 @@ def keyphrase_generator(article_link, model_1, model_2, max_num_keywords, model_
264
 
265
 
266
  igen_pubmed = gr.Interface(keyphrase_generator,
267
- inputs=[gr.inputs.Textbox(lines=1, placeholder="Provide article web link here (Can be chosen from examples below)",default="", label="Article web link"),
268
- gr.inputs.Dropdown(choices=['sentence-transformers/all-mpnet-base-v2',
269
  'sentence-transformers/all-mpnet-base-v1',
270
  'sentence-transformers/all-distilroberta-v1',
271
  'sentence-transformers/gtr-t5-large',
@@ -280,7 +280,7 @@ igen_pubmed = gr.Interface(keyphrase_generator,
280
  type="value",
281
  default='sentence-transformers/stsb-roberta-base-v2',
282
  label="Select any SBERT model for TextRank from the list below"),
283
- gr.inputs.Dropdown(choices=['sentence-transformers/paraphrase-mpnet-base-v2',
284
  'sentence-transformers/all-mpnet-base-v1',
285
  'sentence-transformers/paraphrase-distilroberta-base-v1',
286
  'sentence-transformers/paraphrase-xlm-r-multilingual-v1',
@@ -297,14 +297,14 @@ igen_pubmed = gr.Interface(keyphrase_generator,
297
  type="value",
298
  default='sentence-transformers/all-mpnet-base-v1',
299
  label="Select any SBERT model for keyphrases from the list below"),
300
- gr.inputs.Slider(minimum=5, maximum=20, step=1, default=10, label="Max Keywords"),
301
- gr.inputs.Dropdown(choices=['cambridgeltl/SapBERT-from-PubMedBERT-fulltext',
302
  'cambridgeltl/SapBERT-from-PubMedBERT-fulltext-mean-token'],
303
  type="value",
304
  default='cambridgeltl/SapBERT-from-PubMedBERT-fulltext',
305
  label="Select any SapBERT model for clustering from the list below"),
306
- gr.inputs.Slider(minimum=5, maximum=15, step=1, default=10, label="PubMed Max Abstracts"),
307
- gr.inputs.Dropdown(choices=['pritamdeka/S-Bluebert-snli-multinli-stsb',
308
  'pritamdeka/S-BioBert-snli-multinli-stsb',
309
  'pritamdeka/S-Biomed-Roberta-snli-multinli-stsb',
310
  'sentence-transformers/all-mpnet-base-v2'],
 
264
 
265
 
266
  igen_pubmed = gr.Interface(keyphrase_generator,
267
+ inputs=[gr.components.Textbox(lines=1, placeholder="Provide article web link here (Can be chosen from examples below)",default="", label="Article web link"),
268
+ gr.components.Dropdown(choices=['sentence-transformers/all-mpnet-base-v2',
269
  'sentence-transformers/all-mpnet-base-v1',
270
  'sentence-transformers/all-distilroberta-v1',
271
  'sentence-transformers/gtr-t5-large',
 
280
  type="value",
281
  default='sentence-transformers/stsb-roberta-base-v2',
282
  label="Select any SBERT model for TextRank from the list below"),
283
+ gr.components.Dropdown(choices=['sentence-transformers/paraphrase-mpnet-base-v2',
284
  'sentence-transformers/all-mpnet-base-v1',
285
  'sentence-transformers/paraphrase-distilroberta-base-v1',
286
  'sentence-transformers/paraphrase-xlm-r-multilingual-v1',
 
297
  type="value",
298
  default='sentence-transformers/all-mpnet-base-v1',
299
  label="Select any SBERT model for keyphrases from the list below"),
300
+ gr.components.Slider(minimum=5, maximum=20, step=1, default=10, label="Max Keywords"),
301
+ gr.components.Dropdown(choices=['cambridgeltl/SapBERT-from-PubMedBERT-fulltext',
302
  'cambridgeltl/SapBERT-from-PubMedBERT-fulltext-mean-token'],
303
  type="value",
304
  default='cambridgeltl/SapBERT-from-PubMedBERT-fulltext',
305
  label="Select any SapBERT model for clustering from the list below"),
306
+ gr.components.Slider(minimum=5, maximum=15, step=1, default=10, label="PubMed Max Abstracts"),
307
+ gr.components.Dropdown(choices=['pritamdeka/S-Bluebert-snli-multinli-stsb',
308
  'pritamdeka/S-BioBert-snli-multinli-stsb',
309
  'pritamdeka/S-Biomed-Roberta-snli-multinli-stsb',
310
  'sentence-transformers/all-mpnet-base-v2'],