fix for preview images

special/dl_display_results.py

@@ -165,7 +165,7 @@ display(HTML(CSS))
 HR = widgets.HTML('<hr>')
 
 header_widget = widgets.HTML(value='''
-<div class="header_dl">DOWNLOAD RESULTS <span style="color: grey; opacity: 0.25;">| v0.2.6</span></div>
+<div class="header_dl">DOWNLOAD RESULTS <span style="color: grey; opacity: 0.25;">| v0.2.7</span></div>
 ''')
 
 # Adding content to "outputs_widgets_list"
@@ -176,13 +176,13 @@ def output_container_generator(header, items):
     return container_widget
 
 # Models
-models_list = [file for file in os.listdir(models_dir) if not file.endswith('.txt')]
+models_list = [file for file in os.listdir(models_dir) if file.endswith('.safetensors')]
 models_widget = output_container_generator('Models', models_list).add_class("model")
 # Vaes
-vaes_list = [file for file in os.listdir(vaes_dir) if not file.endswith('.txt')]
+vaes_list = [file for file in os.listdir(vaes_dir) if file.endswith('.safetensors')]
 vaes_widget = output_container_generator('VAEs', vaes_list).add_class("vae")
 # Embeddings
-embeddings_list = [file for file in os.listdir(embeddings_dir) if not file.endswith('.txt') and not os.path.isdir(os.path.join(embeddings_dir, file))]
+embeddings_list = [file for file in os.listdir(embeddings_dir) if file.endswith(('.safetensors', '.pt')) and not os.path.isdir(os.path.join(embeddings_dir, file))]
 embeddings_widget = output_container_generator('Embeddings', embeddings_list).add_class("embedding")
 # LoRAs
 # loras_list = [file for file in os.listdir(loras_dir) if not file.endswith('.txt')]