Update app.py

app.py

@@ -1,7 +1,16 @@
 from youtube_video import download_youtube_video
 import requests
+from huggingface_hub import InferenceClient
+
 import gradio as gr
 data = 'https://www.youtube.com/watch?v=ETDEuH3YL7I' #'https://www.youtube.com/watch?v=bJ5FDtgOwjo'
+
+
+
+client = InferenceClient(model="http://127.0.0.1:8080")
+client.text_generation(prompt="Write a code for snake game")
+
+
 def app(video_link):
     video_path = download_youtube_video(video_link)
     return video_path
@@ -12,7 +21,9 @@ interface = gr.Interface(
     inputs=gr.Textbox(data, label="Enter YouTube link"),
     outputs=gr.Video(label = "video_path")
 )
-gr.Interface.queue(api_open=True)
+#gr.Interface.queue(api_open=True)
+for token in client.text_generation("How do you make cheese?", max_new_tokens=12, stream=True):
+    print(token)
 interface.launch(debug=True)