Update app.py

app.py

@@ -9,7 +9,7 @@ AUTH_TOKEN = os.getenv('AUTH_TOKEN')
 
 def transcribe_audio(file_path):
     if not ASR_API_URL or not AUTH_TOKEN:
-        return "Error: Missing ASR_API_URL or AUTH_TOKEN."
+        return "Error: Missing ASR_API_URL or AUTH_TOKEN.", None
 
     # Prepare headers and data
     headers = {
@@ -22,28 +22,27 @@ def transcribe_audio(file_path):
     start_time = time.time()
     # Send POST request
     response = requests.post(ASR_API_URL, headers=headers, files=files)
-    inference_time = time.time() - start_time # in seconds
+    inference_time = time.time() - start_time  # in seconds
 
     # Check if response is successful
     if response.status_code == 200:
-        return response.json().get("transcription", "No transcription returned.")
+        transcription = response.json().get("transcription", "No transcription returned.")
+        inference_time_str = f"{response.json().get("time", "No inference time returned.")} seconds"
+        return transcription, inference_time_str
     else:
-        return f"Error: {response.status_code}, {response.text}"
-
-
-description_text = """
-The Gooya Persian ASR model is crazy fast and super [powerful](https://huggingface.co/spaces/navidved/open_persian_asr_leaderboard) when it comes to Persian ASR!
-\nJust drop in a Persian audio file, and boom—we’ll hit you with the best transcription you can get! 🚀🔥
-"""
+        return f"Error: {response.status_code}, {response.text}", None
 
 
 # Set up the Gradio interface
 gr.Interface(
     fn=transcribe_audio,
-    inputs=gr.Audio(type="filepath"),  # Updated here
-    outputs="text",
+    inputs=gr.Audio(type="filepath"),
+    outputs=[
+        gr.Textbox(label="Transcription"),
+        gr.Textbox(label="Inference Time")
+    ],
     title="Gooya v1.4 Persian ASR",
-    description= """
+    description="""
 The Gooya Persian ASR model is crazy fast and super [powerful](https://huggingface.co/spaces/navidved/open_persian_asr_leaderboard) when it comes to Persian ASR!
     \nJust drop in a Persian audio file, and boom—we’ll hit you with the best transcription you can get! 🚀🔥
 """