salomonsky commited on
Commit
e8d4e9f
1 Parent(s): 82228c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,9 +5,10 @@ from gtts import gTTS
5
  from pydub import AudioSegment
6
  from transformers import GPT2LMHeadModel, GPT2Tokenizer, pipeline
7
 
 
8
  os.environ["TOKENIZERS_PARALLELISM"] = "true"
9
  generator = pipeline('text-generation', model="salomonsky/deepSP")
10
- tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
11
 
12
  def generate_output(text):
13
  prompt = ""
 
5
  from pydub import AudioSegment
6
  from transformers import GPT2LMHeadModel, GPT2Tokenizer, pipeline
7
 
8
+ model = GPT2LMHeadModel.from_pretrained("salomonsky/deepSP")
9
  os.environ["TOKENIZERS_PARALLELISM"] = "true"
10
  generator = pipeline('text-generation', model="salomonsky/deepSP")
11
+ tokenizer = GPT2Tokenizer.from_pretrained('salomonsky/deepSP')
12
 
13
  def generate_output(text):
14
  prompt = ""