Update app.py

app.py

@@ -6,8 +6,10 @@ import torch
 from transformers import pipeline
 
 # Pre-Initialize
-DEVICE = -1  # -1 indicates CPU for transformers pipeline
-print("[SYSTEM] | Using CPU type compute device.")
+DEVICE = "auto"
+if DEVICE == "auto":
+    DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+print(f"[SYSTEM] | Using {DEVICE} type compute device.")
 
 # Variables
 DEFAULT_TASK = "transcribe"
@@ -24,27 +26,18 @@ footer {
 '''
 
 # Functions
+@spaces.GPU(duration=60)
 def transcribe(input=None, task=DEFAULT_TASK):
     print(input)
-    if input is None:
-        raise gr.Error("Invalid input.")
+    if input is None: raise gr.Error("Invalid input.")
         
-    output = repo(
-        input,
-        batch_size=BATCH_SIZE,
-        generate_kwargs={"task": task},
-        return_timestamps=True
-    )["text"]
+    output = repo(input, batch_size=BATCH_SIZE, generate_kwargs={"task": task}, return_timestamps=True)["text"]
     
     return output
 
 def cloud():
     print("[CLOUD] | Space maintained.")
 
-@spaces.GPU(duration=60)
-def gpu():
-    return
-
 # Initialize
 with gr.Blocks(css=css) as main:
     with gr.Column():