Commit Test SRT

app.py

@@ -1,10 +1,27 @@
-import gradio as gr
+from transformers import pipeline
 
+def summarize_srt(srt_text, max_length=100):
+    # Extract text from SRT format
+    lines = srt_text.split('\n\n')
+    text = ' '.join(line.split('\n')[2] for line in lines if line.strip())
 
-def greet(name):
-    return "Hello " + name
+    # Initialize the summarization pipeline
+    summarizer = pipeline("summarization")
 
+    # Summarize the text
+    summary = summarizer(text, max_length=max_length)
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+    return summary[0]['summary_text']
 
-demo.launch()
+def main():
+    # Take SRT content as input
+    srt_content = input("Enter the SRT content:\n")
+
+    max_length = 100  # Maximum length of the summary
+
+    summary = summarize_srt(srt_content, max_length)
+    print("Summary:")
+    print(summary)
+
+if __name__ == "__main__":
+    main()