Shahadbal commited on
Commit
896f1c2
1 Parent(s): 56e5e27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -51,6 +51,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
51
  # Load the tokenizer and model
52
  # tokenizer = AutoTokenizer.from_pretrained("valhalla/t5-small-qg-prepend", use_fast=False) # Disable the fast tokenizer
53
  # model = AutoModelForSeq2SeqLM.from_pretrained("valhalla/t5-small-qg-prepend")
 
54
 
55
  question_generator = pipeline("question-generation", model="valhalla/t5-small-qg-prepend", qg_format="prepend")
56
 
 
51
  # Load the tokenizer and model
52
  # tokenizer = AutoTokenizer.from_pretrained("valhalla/t5-small-qg-prepend", use_fast=False) # Disable the fast tokenizer
53
  # model = AutoModelForSeq2SeqLM.from_pretrained("valhalla/t5-small-qg-prepend")
54
+ tokenizer = AutoTokenizer.from_pretrained("valhalla/t5-small-qg-prepend", use_fast=False)
55
 
56
  question_generator = pipeline("question-generation", model="valhalla/t5-small-qg-prepend", qg_format="prepend")
57