ParthCodes commited on
Commit
9140841
·
verified ·
1 Parent(s): 77ccdf4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -1
app.py CHANGED
@@ -2,12 +2,22 @@ import json
2
  import ffmpeg
3
  from subprocess import run
4
  import gradio as gr
 
 
 
5
  import whisper_timestamped as whisper
6
  from transformers import pipeline
7
 
8
  model = whisper.load_model("small", device="cpu")
9
  sentiment_analysis = pipeline("sentiment-analysis", framework="pt", model="SamLowe/roberta-base-go_emotions", use_fast=True)
10
 
 
 
 
 
 
 
 
11
  def analyze_sentiment(text):
12
  results = sentiment_analysis(text)
13
  sentiment_results = {result['label']: result['score'] for result in results}
@@ -24,7 +34,9 @@ def transcribe(audio):
24
 
25
 
26
  def video_to_audio(input_video, output_audio):
27
- audio_file = f"test.wav"
 
 
28
  run(["ffmpeg", "-i", 'test_video_1.mp4', audio_file])
29
 
30
  response = transcribe(audio=audio_file)
 
2
  import ffmpeg
3
  from subprocess import run
4
  import gradio as gr
5
+ import uuid
6
+ import os
7
+ from zipfile import ZipFile
8
  import whisper_timestamped as whisper
9
  from transformers import pipeline
10
 
11
  model = whisper.load_model("small", device="cpu")
12
  sentiment_analysis = pipeline("sentiment-analysis", framework="pt", model="SamLowe/roberta-base-go_emotions", use_fast=True)
13
 
14
+ ZipFile("ffmpeg.zip").extractall()
15
+ st = os.stat('ffmpeg')
16
+ os.chmod('ffmpeg', st.st_mode | stat.S_IEXEC)
17
+
18
+ print("cwd", os.getcwd())
19
+ print(os.listdir())
20
+
21
  def analyze_sentiment(text):
22
  results = sentiment_analysis(text)
23
  sentiment_results = {result['label']: result['score'] for result in results}
 
34
 
35
 
36
  def video_to_audio(input_video, output_audio):
37
+ current_path = os.getcwd()
38
+ common_uuid = uuid.uuid4()
39
+ audio_file = f"{common_uuid}.wav"
40
  run(["ffmpeg", "-i", 'test_video_1.mp4', audio_file])
41
 
42
  response = transcribe(audio=audio_file)