From 8fbfb2612dcfd6940e18b4e261ae7519638d6124 Mon Sep 17 00:00:00 2001 From: "ivan.pavlovic" <ivan.pavlovic@hes-so.ch> Date: Fri, 6 Dec 2024 11:12:14 +0100 Subject: [PATCH] Change model file name --- api/model/{facebook.py => facebookBartLargeMnli.py} | 0 ...facebook_results.txt => facebookBartLargeMnli_results.txt} | 0 api/pubmedApi.py | 4 ++-- 3 files changed, 2 insertions(+), 2 deletions(-) rename api/model/{facebook.py => facebookBartLargeMnli.py} (100%) rename api/model/{facebook_results.txt => facebookBartLargeMnli_results.txt} (100%) 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 22fc74c6c..3a63534a0 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("--------------------------------------------------------") -- GitLab