Update app.py

app.py

@@ -314,41 +314,40 @@ igen_pubmed = gr.Interface(keyphrase_generator,
              #outputs=gr.outputs.Dataframe(type="auto", label="Retrieved Results from PubMed",max_cols=2, overflow_row_behaviour="paginate"), 
              outputs=gr.components.JSON(label="Title and Abstracts"),
              #outputs=gr.outputs.File(label=None),
-             theme="peach", layout="horizontal",
              title="PubMed Abstract Retriever", description="Retrieves relevant PubMed abstracts for an online article which can be used as further references. The output is in the form of JSON with <b><i>Title</i></b> and <b><i>Abstract</i></b> as the fields of the JSON output. Please note that it may take sometime for the models to load. Examples are provided below for demo purposes. Choose any one example to see the results. The models can be changed to see different results. ",
-             #examples=[
-                         #["https://www.cancer.news/2021-12-22-mrna-vaccines-weaken-immune-system-cause-cancer.html",                        
-                         #'sentence-transformers/all-mpnet-base-v1',
-                         #'sentence-transformers/paraphrase-MiniLM-L12-v2',
-                         #10,
-                         #'cambridgeltl/SapBERT-from-PubMedBERT-fulltext',
-                         #15,
-                         #'pritamdeka/S-Biomed-Roberta-snli-multinli-stsb'],
+             examples=[
+                         ["https://www.cancer.news/2021-12-22-mrna-vaccines-weaken-immune-system-cause-cancer.html",                        
+                         'sentence-transformers/all-mpnet-base-v1',
+                         'sentence-transformers/paraphrase-MiniLM-L12-v2',
+                         10,
+                         'cambridgeltl/SapBERT-from-PubMedBERT-fulltext',
+                         15,
+                         'pritamdeka/S-Biomed-Roberta-snli-multinli-stsb'],
                          
-                         #["https://www.cancer.news/2022-02-04-doctors-testifying-covid-vaccines-causing-cancer-aids.html#",                        
-                         #'sentence-transformers/all-mpnet-base-v1',
-                         #'sentence-transformers/all-mpnet-base-v1',
-                         #12,
-                         #'cambridgeltl/SapBERT-from-PubMedBERT-fulltext',
-                         #11,
-                         #'pritamdeka/S-Biomed-Roberta-snli-multinli-stsb'],
+                         ["https://www.cancer.news/2022-02-04-doctors-testifying-covid-vaccines-causing-cancer-aids.html#",                        
+                         'sentence-transformers/all-mpnet-base-v1',
+                         'sentence-transformers/all-mpnet-base-v1',
+                         12,
+                         'cambridgeltl/SapBERT-from-PubMedBERT-fulltext',
+                         11,
+                         'pritamdeka/S-Biomed-Roberta-snli-multinli-stsb'],
                          
-                         #["https://www.medicalnewstoday.com/articles/alzheimers-addressing-sleep-disturbance-may-alleviate-symptoms",
-                         #'pritamdeka/S-Biomed-Roberta-snli-multinli-stsb',
-                         #'sentence-transformers/all-mpnet-base-v1',
-                         #10,
-                         #'cambridgeltl/SapBERT-from-PubMedBERT-fulltext',
-                         #10,
-                         #'pritamdeka/S-Biomed-Roberta-snli-multinli-stsb'],
+                         ["https://www.medicalnewstoday.com/articles/alzheimers-addressing-sleep-disturbance-may-alleviate-symptoms",
+                         'pritamdeka/S-Biomed-Roberta-snli-multinli-stsb',
+                         'sentence-transformers/all-mpnet-base-v1',
+                         10,
+                         'cambridgeltl/SapBERT-from-PubMedBERT-fulltext',
+                         10,
+                         'pritamdeka/S-Biomed-Roberta-snli-multinli-stsb'],
                          
-                        #["https://www.medicalnewstoday.com/articles/omicron-what-do-we-know-about-the-stealth-variant",
-                        # 'pritamdeka/S-Biomed-Roberta-snli-multinli-stsb',
-                        # 'sentence-transformers/all-mpnet-base-v1',
-                        # 15,
-                        # 'cambridgeltl/SapBERT-from-PubMedBERT-fulltext',
-                        # 10,
-                        # 'pritamdeka/S-Biomed-Roberta-snli-multinli-stsb']                       
-                       #],                     
+                        ["https://www.medicalnewstoday.com/articles/omicron-what-do-we-know-about-the-stealth-variant",
+                         'pritamdeka/S-Biomed-Roberta-snli-multinli-stsb',
+                         'sentence-transformers/all-mpnet-base-v1',
+                         15,
+                         'cambridgeltl/SapBERT-from-PubMedBERT-fulltext',
+                         10,
+                         'pritamdeka/S-Biomed-Roberta-snli-multinli-stsb']                       
+                       ],                     
              article= "This work is based on the paper <a href=https://dl.acm.org/doi/10.1145/3487664.3487701>provided here</a>." 
              "\t It uses the TextRank algorithm with SBERT to first find the top sentences and then extracts the keyphrases from those sentences using scispaCy and SBERT."
              "\t The application then uses a UMLS based BERT model, <a href=https://arxiv.org/abs/2010.11784>SapBERT</a> to cluster the keyphrases using K-means clustering method and finally create a boolean query. After that the top k titles and abstracts are retrieved from PubMed database and displayed according to relevancy. The SapBERT models can be changed as per the list provided. "