Spaces:

calvininterview
/

bart-question-interactive

Runtime error

haramkoo commited on Mar 7, 2022

Commit

93e8a0e

1 Parent(s): a9c4623

webapp update 03-07-2022 6:17 pm

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ model = BartForConditionalGeneration.from_pretrained("hyechanjun/interview-quest
 tok = BartTokenizer.from_pretrained("hyechanjun/interview-question-remake")
 def genQuestion(context):
-    inputs = tok.tokenize(context, return_tensors="pt")
     output = model.generate(inputs["input_ids"], num_beams=4, max_length=64)
     return tok.batch_decode(output, skip_special_tokens=True, clean_up_tokenization_spaces=False)

 tok = BartTokenizer.from_pretrained("hyechanjun/interview-question-remake")
 def genQuestion(context):
+    inputs = tok(context, return_tensors="pt")
     output = model.generate(inputs["input_ids"], num_beams=4, max_length=64)
     return tok.batch_decode(output, skip_special_tokens=True, clean_up_tokenization_spaces=False)