Update app.py

app.py

@@ -10,7 +10,6 @@ import os
 import warnings
 from pydub import AudioSegment
 import torch
-from torch import torchaudio
 from transformers import pipeline, WhisperForConditionalGeneration, WhisperProcessor
 from huggingface_hub import model_info
 import nltk
@@ -44,7 +43,7 @@ def convert_to_wav(audio_file):
     return wav_file
 
 # Initialize device for torch
-#device = "cuda" if torch.cuda.is_available() else "cpu"
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Load smoking-big-pipe
 MODEL_NAME = "NbAiLab/nb-whisper-large"
@@ -185,15 +184,22 @@ def save_to_pdf(text, summary):
     pdf.output(pdf_output_path)
     return pdf_output_path
 
+def _return_img_html_embed(img_url):
+    HTML_str = (
+        f'<center> <img src="{img_url}" alt="Image" width="100%" height="auto"> </center>'
+    )
+    return HTML_str
+
 # Gradio Interface
+def display_image():
+    img_url = "https://huggingface.co/spaces/camparchimedes/transcription_app/raw/main/banner_trans.png"
+    html_embed_str = _return_img_html_embed(img_url)
+    return html_embed_str
+
 iface = gr.Blocks()
 
 with iface:
-    gr.HTML("""
-    <div style="text-align: center;">
-        <img src="https://huggingface.co/spaces/camparchimedes/transcription_app/raw/main/banner_trans.png" alt="" width="100%" height="auto">
-    </div>
-    """)
+    gr.HTML(display_image())
     gr.Markdown("# Vi har nå muligheten til å oversette lydfiler til norsk skrift.")
 
     with gr.Tabs():