diff --git a/api/model/facebook.py b/api/model/facebookBartLargeMnli.py similarity index 100% rename from api/model/facebook.py rename to api/model/facebookBartLargeMnli.py diff --git a/api/model/facebook_results.txt b/api/model/facebookBartLargeMnli_results.txt similarity index 100% rename from api/model/facebook_results.txt rename to api/model/facebookBartLargeMnli_results.txt diff --git a/api/pubmedApi.py b/api/pubmedApi.py index 22fc74c6c61e3f018de77688f32cbec08f9befce..3a63534a0295e0139fe9f7b8feb4ab3a63de711d 100644 --- a/api/pubmedApi.py +++ b/api/pubmedApi.py @@ -1,6 +1,6 @@ from requests import get from parser.xmlParser import parseXmlFile -import model.facebook as facebook +import model.facebookBartLargeMnli as facebookBartLargeMnli FILENAME = "pubmedData.xml" @@ -61,7 +61,7 @@ for key in obj.keys(): print(meshTerms) - facebook.classify(title + " \n " + articleTitle + " \n " + abstract) + facebookBartLargeMnli.classify(title + " \n " + articleTitle + " \n " + abstract) else: print(f"It has no abstract: {entrie["MedlineCitation"]["PMID"]}") print("--------------------------------------------------------")