Update app.py

app.py

@@ -1,22 +1,22 @@
 import gradio as gr
 import pytube
 from transformers import pipeline
-# from time import sleep
+from time import sleep
 
 
 generator = pipeline('text-generation', model='parseny/youtube_comment_generation')
 
 def generate(text):
     yt = pytube.YouTube(text)
-    # while True:
-    #     try:
-    #         title = yt.title
-    #         break
-    #     except:
-    #         sleep(1)
-    #         yt = pytube.YouTube(text)
-    #         continue
-    title = yt.title
+    while True:
+        try:
+            title = yt.title
+            break
+        except:
+            sleep(1)
+            yt = pytube.YouTube(text)
+            continue
+    # title = yt.title
     title = title.lower()
     title = title.replace("[^a-zA-Z#]", " ")
     title = title.replace(" +", " ")