Update generate_model_grid.py

generate_model_grid.py

@@ -10,8 +10,8 @@ import os
 
 import modules.scripts as scripts
 import gradio as gr
-
 from modules import images, sd_samplers
+from modules.paths import models_path
 from modules.hypernetworks import hypernetwork
 from modules.processing import process_images, Processed, StableDiffusionProcessingTxt2Img
 from modules.shared import opts, cmd_opts, state
@@ -241,8 +241,7 @@ class Script(scripts.Script):
 
     def ui(self, is_img2img):
         filenames = []
-        dirpath = '/content/stable-diffusion-webui/models/Stable-diffusion/'
-        for path in os.listdir(dirpath):
+        for path in os.listdir(os.path.join(models_path, 'Stable-diffusion')):
             if path.endswith('.ckpt'):
                 filenames.append(path)