Update app.py

app.py

@@ -1,18 +1,18 @@
 import re
 from youtube_transcript_api import YouTubeTranscriptApi
 from youtube_transcript_api.formatters import TextFormatter
-import torch
-import gradio as gr
-from transformers import pipeline
+from gradio_client import Client
 
-text_summary = pipeline("summarization", model="dicta-il/dictalm2.0", torch_dtype=torch.bfloat16)
+# Initialize the Gradio client with your app's name
+client = Client("Hwilner/hebrew_summerizer")
 
-
-
-
-def summary (input):
-    output = text_summary(input)
-    return output[0]['summary_text']
+def summary(input_text):
+    # Use the API endpoint to get the summary
+    result = client.predict(
+        input_text,
+        api_name="/predict"
+    )
+    return result
 
 def extract_video_id(url):
     # Regex to extract the video ID from various YouTube URL formats
@@ -22,7 +22,6 @@ def extract_video_id(url):
         return match.group(1)
     return None
 
-
 def get_youtube_transcript(video_url):
     video_id = extract_video_id(video_url)
     if not video_id:
@@ -35,23 +34,19 @@ def get_youtube_transcript(video_url):
         # Format the transcript into plain text
         formatter = TextFormatter()
         text_transcript = formatter.format_transcript(transcript)
+        
+        # Get the summary from the API
         summary_text = summary(text_transcript)
 
         return summary_text
     except Exception as e:
         return f"An error occurred: {e}"
 
-
-# Example URL (Replace this with the actual URL when using the script)
-# video_url = "https://youtu.be/5PibknhIsTc"
-# print(get_youtube_transcript(video_url))
-
 gr.close_all()
 
-# demo = gr.Interface(fn=summary, inputs="text",outputs="text")
 demo = gr.Interface(fn=get_youtube_transcript,
-                    inputs=[gr.Textbox(label="Input YouTube Url to summarize",lines=1)],
-                    outputs=[gr.Textbox(label="Summarized text",lines=4)],
-                    title=" YouTube  Script Summarizer-Hebrew",
+                    inputs=[gr.Textbox(label="Input YouTube Url to summarize", lines=1)],
+                    outputs=[gr.Textbox(label="Summarized text", lines=4)],
+                    title="YouTube Script Summarizer - Hebrew",
                     description="THIS APPLICATION WILL BE USED TO SUMMARIZE THE YOUTUBE VIDEO SCRIPT.")
 demo.launch()