Update app.py

app.py

@@ -6,10 +6,10 @@ import gradio as gr
 from bs4 import BeautifulSoup
 import requests
 
-"""
+
 def model_url_list():
     url_list = []
-    for i in range(0, 3):
+    for i in range(0, 5):
         url_list.append(f"https://huggingface.co/models?other=stable-diffusion&p={i}&sort=downloads")
     return url_list
 
@@ -27,7 +27,7 @@ def data_scraping(url_list):
 model_list = data_scraping(model_url_list())
 for i in range(len(model_list)):
     model_list[i] = model_list[i][1:]
-"""
+
 best_model_list = [
     "runwayml/stable-diffusion-v1-5",
     "CompVis/stable-diffusion-v1-4",
@@ -38,7 +38,7 @@ best_model_list = [
     "sd-dreambooth-library/disco-diffusion-style"
 ]
 
-model_list = best_model_list
+model_list = best_model_list + model_list
 
 def tune_video_predict(
     pipe_id: str,