Update app.py

app.py

@@ -3,6 +3,8 @@ import gradio as gr
 import os
 import requests
 from transformers import pipeline
+import torch
+device = 0 if torch.cuda.is_available() else -1
 
 # Set your FastAPI backend endpoint
 BACKEND_URL = "https://asr-evaluation-backend.emergentai.ug/submit-feedback"
@@ -33,7 +35,7 @@ os.makedirs("responses", exist_ok=True)
 
 # Transcription function
 def transcribe(audio, language):
-    asr = pipeline("automatic-speech-recognition", model=model_map[language], device=0)
+    asr = pipeline("automatic-speech-recognition", model=model_map[language], device)
     text = asr(audio)["text"]
     return text, audio
 
@@ -136,6 +138,6 @@ with gr.Blocks() as demo:
             else:
                 return gr.update(visible=False)
     
-        proceed_btn.click(fn=reveal_ui, inputs=[user_id], outputs=[main_ui])
+    proceed_btn.click(fn=reveal_ui, inputs=[user_id], outputs=[main_ui])
 # Launch the interface
 demo.launch()