fix widget

create_handler.ipynb

@@ -151,7 +151,7 @@
     "        result = self.model.transcribe(audio_nparray)\n",
     "\n",
     "        # postprocess the prediction\n",
-    "        return {\"transcription\": result[\"text\"]}"
+    "        return {\"text\": result[\"text\"]}"
    ]
   },
   {
@@ -163,7 +163,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 6,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -175,7 +175,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {

handler.py

@@ -30,4 +30,4 @@ class EndpointHandler():
         result = self.model.transcribe(audio_nparray)
 
         # postprocess the prediction
-        return {"transcription": result["text"]}
+        return {"text": result["text"]}