Create pipeline_config.json

pipeline_config.json

@@ -0,0 +1,25 @@
+pipeline_config = {
+    "task": "text-generation",
+    "framework": "pt",
+    "model": {
+        "name": "ombhojane/wellness-mini",
+        "type": "AutoModelForCausalLM"
+    },
+    "tokenizer": {
+        "name": "ombhojane/wellness-mini",
+        "type": "AutoTokenizer"
+    }
+}
+
+# Save pipeline config
+import json
+with open("pipeline_config.json", "w") as f:
+    json.dump(pipeline_config, f, indent=2)
+
+# Upload pipeline config
+api.upload_file(
+    path_or_fileobj="pipeline_config.json",
+    path_in_repo="pipeline_config.json",
+    repo_id="ombhojane/wellness-mini",
+    repo_type="model"
+)