Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -6,10 +6,10 @@ import gradio as gr
|
|
6 |
from bs4 import BeautifulSoup
|
7 |
import requests
|
8 |
|
9 |
-
|
10 |
def model_url_list():
|
11 |
url_list = []
|
12 |
-
for i in range(0,
|
13 |
url_list.append(f"https://huggingface.co/models?other=stable-diffusion&p={i}&sort=downloads")
|
14 |
return url_list
|
15 |
|
@@ -27,7 +27,7 @@ def data_scraping(url_list):
|
|
27 |
model_list = data_scraping(model_url_list())
|
28 |
for i in range(len(model_list)):
|
29 |
model_list[i] = model_list[i][1:]
|
30 |
-
|
31 |
best_model_list = [
|
32 |
"runwayml/stable-diffusion-v1-5",
|
33 |
"CompVis/stable-diffusion-v1-4",
|
@@ -38,7 +38,7 @@ best_model_list = [
|
|
38 |
"sd-dreambooth-library/disco-diffusion-style"
|
39 |
]
|
40 |
|
41 |
-
model_list = best_model_list
|
42 |
|
43 |
def tune_video_predict(
|
44 |
pipe_id: str,
|
|
|
6 |
from bs4 import BeautifulSoup
|
7 |
import requests
|
8 |
|
9 |
+
|
10 |
def model_url_list():
|
11 |
url_list = []
|
12 |
+
for i in range(0, 5):
|
13 |
url_list.append(f"https://huggingface.co/models?other=stable-diffusion&p={i}&sort=downloads")
|
14 |
return url_list
|
15 |
|
|
|
27 |
model_list = data_scraping(model_url_list())
|
28 |
for i in range(len(model_list)):
|
29 |
model_list[i] = model_list[i][1:]
|
30 |
+
|
31 |
best_model_list = [
|
32 |
"runwayml/stable-diffusion-v1-5",
|
33 |
"CompVis/stable-diffusion-v1-4",
|
|
|
38 |
"sd-dreambooth-library/disco-diffusion-style"
|
39 |
]
|
40 |
|
41 |
+
model_list = best_model_list + model_list
|
42 |
|
43 |
def tune_video_predict(
|
44 |
pipe_id: str,
|