Spaces:

parseny
/

youtube_comment_generation

Runtime error

parseny commited on Apr 16, 2023

Commit

d07c980

•

1 Parent(s): 7339057

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 import pytube
-generator = pipeline('text-generation', model='parseny/youtube_comments')
 def generate(text):
     yt = pytube.YouTube(text)

 from transformers import pipeline
 import pytube
+generator = pipeline('text-generation', model='parseny/youtube_comment_generation_02')
 def generate(text):
     yt = pytube.YouTube(text)