Update model_index.json

model_index.json

@@ -1,27 +1,29 @@
 {
-  "_class_name": "StableDiffusionPipeline",
   "model_type": "ImageGeneration",
-  "base_model": "UNet",
-  "scheduler": "DDIMScheduler",
   "components": {
     "unet": {
       "_class_name": "UNet2DConditionModel",
+      "library": "diffusers",
       "file": "unet/unet.bin"
     },
     "vae": {
       "_class_name": "AutoencoderKL",
+      "library": "diffusers",
       "file": "vae/vae.bin"
     },
     "scheduler": {
       "_class_name": "DDIMScheduler",
+      "library": "diffusers",
       "file": "scheduler/scheduler.json"
     },
     "text_encoder": {
       "_class_name": "CLIPTextModel",
+      "library": "transformers",
       "file": "text_encoder/text_encoder.bin"
     },
     "tokenizer": {
       "_class_name": "CLIPTokenizer",
+      "library": "transformers",
       "file": "tokenizer/tokenizer.json"
     }
   }