Update app.py

app.py

@@ -5,10 +5,11 @@ import torch
 import gradio as gr
 from transformers import pipeline
 
-text_summary = pipeline("summarization", model="Falconsai/text_summarization", torch_dtype=torch.bfloat16)
+text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
+text_translate = pipeline("translation", model="google-t5/t5-small", torch_dtype=torch.bfloat16)
 
 def summary (input):
-    max_length = 1024  # adjust this value as needed
+    max_length = 2048  # adjust this value as needed
     if len(input) > max_length:
         input = input[:max_length]
     output = text_summary(input)
@@ -45,7 +46,14 @@ def get_german_youtube_transcript(video_url):
         formatter = TextFormatter()
         text_transcript = formatter.format_transcript(transcript)
         summary_text = summary(text_transcript)
-        return summary_text
+        
+        summary_text = summary_text[:300] 
+        #text_transcript = text_transcript[:300]  # truncate to 300 characters
+
+        #transcript_output = text_translate(text_transcript)
+        #transcript_output = transcript_output[0]['translation_text']
+        summary_output = text_translate(summary_text)
+        return summary_output[0]['translation_text'], text_transcript
     except Exception as e:
         return f"Es ist ein Fehler aufgetreten: {e}"
 
@@ -53,9 +61,12 @@ gr.close_all()
 
 demo = gr.Interface(fn=get_german_youtube_transcript,
                     inputs=[gr.Textbox(label="YouTube Url zum Zusammenfassen eingeben",lines=1)],
-                    outputs=[gr.Textbox(label="Zusammengefasster Text",lines=4)],
-                    title="Projekt 2: YouTube-Skript-Zusammenfassung",
-                    description="DIESE ANWENDUNG WIRD VERWENDET, UM DAS SKRIPT EINES YOUTUBE-VIDEOS ZUSAMMENZUFASSEN.",
+                    outputs=[
+                        gr.Textbox(label="Zusammengefasster Text (DE)",lines=4),
+                        gr.Textbox(label="Video-Transkript (EN)", lines=10) 
+                    ],
+                    title="Projekt 2: YouTube-Video(EN)-Zusammenfassung (DE)",
+                    description="Diese Anwendung wird genutzt, um den Inhalt eines YouTube-Videos auf Englisch in eine deutsche Zusammenfassung zu übersetzen.",
                     allow_flagging="never",
                     submit_btn="Übermitteln",
                     clear_btn="Bereinigen",