Hugging Face's logo

Spaces:
dantedgp
/
question-generator
Sleeping

App Files Community
dantedgp commited on
Commit
c286457
·
1 Parent(s): fc48b53

Download NLTK punkt

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -2,6 +2,8 @@ import streamlit as st
2
  import nltk
3
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
4
 
 
 
5
  # Instantiate the model
6
  model = AutoModelForSeq2SeqLM.from_pretrained("dantedgp/question-generator")
7
  tokenizer = AutoTokenizer.from_pretrained("google-t5/t5-small")
 
2
  import nltk
3
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
4
 
5
+ # Initialize NLTK
6
+ nltk.download('punkt')
7
  # Instantiate the model
8
  model = AutoModelForSeq2SeqLM.from_pretrained("dantedgp/question-generator")
9
  tokenizer = AutoTokenizer.from_pretrained("google-t5/t5-small")