From 486bff9d126a7400eb205920f666121b2d9fb8ce Mon Sep 17 00:00:00 2001
From: LudwigForsberg <ludfo774@student.liu.se>
Date: Wed, 7 Dec 2022 16:07:01 +0100
Subject: [PATCH] setup remote fetch to model
---
bart/generate_single_query.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/bart/generate_single_query.py b/bart/generate_single_query.py
index 5ade102..4865619 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
--
GitLab