Update: Improve config.json implementation

config.json

@@ -1,22 +1,11 @@
 {
-  "task": "text-to-image",
-  "framework": "custom",
-  "model-type": "text-to-image",
-  "inference": {
+  "task_type": "text-to-image",
+  "inference_framework": "pytorch",
+  "inference_config": {
     "task_type": "text-to-image",
-    "text_to_image": {
-      "inputs": {
-        "text": {
-          "type": "string",
-          "description": "Input text prompt"
-        }
-      },
-      "outputs": {
-        "image": {
-          "type": "image",
-          "description": "Generated image"
-        }
-      }
+    "runtime": "docker",
+    "docker_config": {
+      "use_dockerfile": true
     }
   }
 }