Update app.py

app.py

@@ -16,6 +16,9 @@ class GradioInference():
         self.keyword_model = T5ForConditionalGeneration.from_pretrained("Voicelab/vlt5-base-keywords")
         self.keyword_tokenizer = T5Tokenizer.from_pretrained("Voicelab/vlt5-base-keywords")
 
+        # Sentiment Classifier
+        self.classifier = pipeline("text-classification")
+
     def __call__(self, link, lang, size):
         if self.yt is None:
             self.yt = YouTube(link)
@@ -39,13 +42,16 @@ class GradioInference():
         output = self.keyword_model.generate(input_ids, no_repeat_ngram_size=3, num_beams=4)
         predicted = self.keyword_tokenizer.decode(output[0], skip_special_tokens=True)
         keywords = [x.strip() for x in predicted.split(',') if x.strip()]
+
+        label = self.classifier(results["text"])[0]["label"]
         
-        return results["text"], transcription_summary[0]["summary_text"], keywords
+        return results["text"], transcription_summary[0]["summary_text"], keywords, label
 
     def populate_metadata(self, link):
         self.yt = YouTube(link)
         return self.yt.thumbnail_url, self.yt.title
 
+
 def transcribe_audio(audio_file):
     model = whisper.load_model("base")
     result = model.transcribe(audio_file)
@@ -84,9 +90,10 @@ with block as demo:
                 with gr.Row().style(equal_height=True):
                     summary = gr.Textbox(label="Summary", placeholder="Summary Output", lines=5)
                     keywords = gr.Textbox(label="Keywords", placeholder="Keywords Output", lines=5)
+                    label = gr.Label(label="Sentiment Analysis")
                 with gr.Row().style(equal_height=True):
                     btn = gr.Button("Get video insights")  # Updated button label
-                btn.click(gio, inputs=[link, lang, size], outputs=[text, summary, keywords])
+                btn.click(gio, inputs=[link, lang, size], outputs=[text, summary, keywords, label])
                 link.change(gio.populate_metadata, inputs=[link], outputs=[img, title])
 
         with gr.Tab("From Audio file"):