save

README.md

@@ -1,6 +1,6 @@
 ---
-title: Whisper
-emoji: 📉
+title: V-to-text
+emoji: 🗣
 colorFrom: pink
 colorTo: yellow
 sdk: gradio

app.py

@@ -5,7 +5,7 @@ import whisper
 from flask import Flask, jsonify, request
 import requests
 import streamlit as st
-
+import time
 
 
 model = whisper.load_model("small")
@@ -18,6 +18,8 @@ def indexApi():
 
 @app.route("/run", methods=['POST'])
 def runApi():
+    start_time = time.time()
+
     audio_url = request.form.get("audio_url")
     # key = request.form.get("key")
     # modelSelection = request.form.get("model")
@@ -62,11 +64,16 @@ def runApi():
     
     options = whisper.DecodingOptions(fp16 = False)
     result = whisper.decode(model, mel, options)
-    print(result.text)
+
+
+    end_time = time.time()
+    total_time = end_time - start_time
+
     return jsonify({
         "audio_url": audio_url,
-        "model": model,
-        # "result": result.text,
+        # "model": model,
+        "result": result.text,
+        "exec_time_sec": total_time
     })
 
 if __name__ == "__main__":