diff --git a/bart/generate_single_query.py b/bart/generate_single_query.py index 5ade102d7ee3c81214a932adf1f42fa08389b51f..4865619f981bc0211f813e31f315ad70692d8978 100644 --- a/bart/generate_single_query.py +++ b/bart/generate_single_query.py @@ -6,7 +6,7 @@ from transformers.models.bart.modeling_bart import shift_tokens_right # Save the model and tokenizer as local variables to use them in the predict_query function def setup(): # Import model from ./trained-models/blackbox - model = BartForConditionalGeneration.from_pretrained("./trained-models/blackbox") + model = BartForConditionalGeneration.from_pretrained("ludfo774/sparql-bart-append-1") tokenizer = BartTokenizer.from_pretrained("facebook/bart-base") model.to('cuda') return model, tokenizer diff --git a/templates/index.html b/templates/index.html index e5370b3d83c90c3fa90052212574e16b949c8d1a..a764f7c3a3a248f5b9797ff845177c6d938095d4 100644 --- a/templates/index.html +++ b/templates/index.html @@ -29,7 +29,7 @@ class="input" placeholder="Enter a question" type="text" - >How many religion's people have died in Malaysia?</span + >Who is the wife of Barack Obama?</span > <span class="underline"></span> </div> diff --git a/trained-models/blackbox/pytorch_model.bin b/trained-models/blackbox/pytorch_model.bin index d6ea6f38c8be8b07c44db78ccc812e2fe23aa211..bb2f0ef68f3d38324b4cf72735cc7192527fa2a6 100644 Binary files a/trained-models/blackbox/pytorch_model.bin and b/trained-models/blackbox/pytorch_model.bin differ diff --git a/trained-models/blackbox/training_args.bin b/trained-models/blackbox/training_args.bin index 08720edef375b1a4a07dd8af06e075b1c470a237..7b173464e5fdcd24cb11446ae1a041053cd2b935 100644 Binary files a/trained-models/blackbox/training_args.bin and b/trained-models/blackbox/training_args.bin differ