Update app.py

app.py

@@ -48,12 +48,13 @@ HEADER_INFO = """
     # WEB APP ✨| Norwegian WHISPER Model
 Switch Work [Transkribering av lydfiler til norsk skrift]
 """.strip()
-LOGO = "https://huggingface.co/spaces/camparchimedes/transcription_app/resolve/main/logo.png"
+LOGO = "https://huggingface.co/spaces/camparchimedes/transcription_app/blob/main/logo.png"
 SIDEBAR_INFO = f"""
 <div align="center">
     <img src="{LOGO}" style="width: 100%; height: auto;"/>
 </div>
 """
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 def convert_to_wav(filepath):
     _,file_ending = os.path.splitext(f'{filepath}')
@@ -61,12 +62,9 @@ def convert_to_wav(filepath):
     os.system(f'ffmpeg -i "{filepath}" -ar 16000 -ac 1 -c:a pcm_s16le "{audio_file}"')
     return audio_file
 
-pipe = pipeline("automatic-speech-recognition",  model="NbAiLab/nb-whisper-large",  device=0)
+pipe = pipeline("automatic-speech-recognition",  model="NbAiLab/nb-whisper-large",  device=0 if device == "cuda" else -1)
 
 def transcribe_audio(audio_file):
-    #if audio_file.endswith(".m4a"):
-        #audio_file = convert_to_wav(audio_file)
-
     start_time = time.time()
 
     outputs = pipe(audio_file, return_timestamps=False, generate_kwargs={'task': 'transcribe', 'language': 'no'}) # skip_special_tokens=True
@@ -76,15 +74,24 @@ def transcribe_audio(audio_file):
     
     output_time = end_time - start_time
     word_count = len(text.split())
+
+    # GPU usage
     memory = psutil.virtual_memory()
     gpu_utilization, gpu_memory = GPUInfo.gpu_usage()
     gpu_utilization = gpu_utilization[0] if len(gpu_utilization) > 0 else 0
     gpu_memory = gpu_memory[0] if len(gpu_memory) > 0 else 0
+
+    # CPU usage
+    cpu_usage = psutil.cpu_percent(interval=1)
+
+    # System info string
     system_info = f"""
     *Memory: {memory.total / (1024 * 1024 * 1024):.2f}GB, used: {memory.percent}%, available: {memory.available / (1024 * 1024 * 1024):.2f}GB.*
     *Processing time: {output_time:.2f} seconds.*
     *Number of words: {word_count}*
-    *GPU Utilization: {gpu_utilization}%, GPU Memory: {gpu_memory}*"""
+    *GPU Utilization: {gpu_utilization}%, GPU Memory: {gpu_memory}*
+    *CPU Usage: {cpu_usage}%*
+    """
 
     return text.strip(), system_info
 #:::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::