Change docstring

model.py

@@ -12,7 +12,7 @@ import os
 class PreTrainedModel(Runner):
     def __init__(self, path=""):
         """
-        Loads model and tokenizer from local directory
+        Initialize downstream model.
         """
         ckp_file = os.path.join(path, "hubert_asr.ckpt")
         ckp = torch.load(ckp_file, map_location='cpu')
@@ -41,12 +41,13 @@ class PreTrainedModel(Runner):
             features = self.upstream.model(inputs)
             features = self.featurizer.model(inputs, features)
             preds = self.downstream.model.inference(features, [])
-        return preds[0]
+        return {"text": preds[0]}
 
 
 """
 import subprocess
 import numpy as np
+from datasets import load_dataset
 # This is already done in the Inference API
 def ffmpeg_read(bpayload: bytes, sampling_rate: int) -> np.array:
     ar = f"{sampling_rate}"