parseny commited on
Commit
3fae9c4
·
1 Parent(s): 92379f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  from time import sleep
5
 
6
 
7
- generator = pipeline('text-generation', model='parseny/youtube_comment_generation_01')
8
 
9
  def generate(text):
10
  yt = pytube.YouTube(text)
 
4
  from time import sleep
5
 
6
 
7
+ generator = pipeline('text-generation', model='parseny/youtube_comment_generation_model')
8
 
9
  def generate(text):
10
  yt = pytube.YouTube(text)