Update app.py

app.py

@@ -73,7 +73,7 @@ class GradioInference:
 
         # Perform summarization on the transcription
         transcription_summary = self.summarizer(
-            results["text"], max_length=512, min_length=30, do_sample=False
+            results["text"], max_length=150, min_length=30, do_sample=False
         )
 
         #### Prueba
@@ -123,13 +123,31 @@ class GradioInference:
         # WordCloud image to display
         wordcloud_image = wordcloud.to_image()
 
-        return (
-            results["text"],
-            summary,
-            keywords,
-            label,
-            wordcloud_image,
-        )
+        if lang.lower() == "english":
+            return (
+                results["text"],
+                transcription_summary[0]["summary_text"],
+                keywords,
+                label,
+                wordcloud_image,
+            )
+        else:
+            return (
+                results["text"],
+                summary,
+                keywords,
+                label,
+                wordcloud_image,
+            )
+
+        
+        # return (
+        #     results["text"],
+        #     summary,
+        #     keywords,
+        #     label,
+        #     wordcloud_image,
+        # )
 
     def populate_metadata(self, link):
         """
@@ -161,7 +179,7 @@ class GradioInference:
 
         # Perform summarization on the transcription
         transcription_summary = self.summarizer(
-            results["text"], max_length=512, min_length=30, do_sample=False
+            results["text"], max_length=150, min_length=30, do_sample=False
         )
 
         #### Prueba
@@ -210,13 +228,30 @@ class GradioInference:
         )
         wordcloud_image = wordcloud.to_image()
 
-        return (
-            results["text"],
-            summary,
-            keywords,
-            label,
-            wordcloud_image,
-        )
+        if lang.lower() == "english":
+            return (
+                results["text"],
+                transcription_summary[0]["summary_text"],
+                keywords,
+                label,
+                wordcloud_image,
+            )
+        else:
+            return (
+                results["text"],
+                summary,
+                keywords,
+                label,
+                wordcloud_image,
+            )
+        
+        # return (
+        #     results["text"],
+        #     summary,
+        #     keywords,
+        #     label,
+        #     wordcloud_image,
+        # )
 
 
 gio = GradioInference()
@@ -277,8 +312,7 @@ with block as demo:
                     inputs=[link, lang, size],
                     outputs=[text, summary, keywords, label, wordcloud_image],
                 )
-                if link:
-                    link.change(gio.populate_metadata, inputs=[link], outputs=[img, title])
+                # link.change(gio.populate_metadata, inputs=[link], outputs=[img, title])
 
         with gr.Tab("From Audio file"):
             with gr.Box():