Update model.py

model.py

@@ -119,4 +119,11 @@ class IndicASRModel(PreTrainedModel):
         local_files_only=False,
         revision=None, **kwargs):
         loc = snapshot_download(repo_id=pretrained_model_name_or_path, token=token)
-        return cls(IndicASRConfig(ts_folder=loc))
+        return cls(IndicASRConfig(ts_folder=loc))
+
+if __name__ == '__main__':
+    from transformers import AutoConfig, AutoModel
+
+    # Register the model so it can be used with AutoModel
+    AutoConfig.register("iasr", IndicASRConfig)
+    AutoModel.register(IndicASRConfig, IndicASRModel)