kadirnar commited on
Commit
5664950
1 Parent(s): 6396079

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -6,6 +6,7 @@ import gradio as gr
6
  from bs4 import BeautifulSoup
7
  import requests
8
 
 
9
  def model_url_list():
10
  url_list = []
11
  for i in range(0, 3):
@@ -26,17 +27,18 @@ def data_scraping(url_list):
26
  model_list = data_scraping(model_url_list())
27
  for i in range(len(model_list)):
28
  model_list[i] = model_list[i][1:]
29
-
30
  best_model_list = [
31
  "runwayml/stable-diffusion-v1-5",
 
32
  "prompthero/openjourney",
33
- "stabilityai/stable-diffusion-2",
34
  "dreamlike-art/dreamlike-diffusion-1.0",
35
- "nitrosocke/Nitro-Diffusion",
36
- "CompVis/stable-diffusion-v1-4",
37
  ]
38
 
39
- model_list = best_model_list + model_list
40
 
41
  def tune_video_predict(
42
  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, 3):
 
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",
34
  "prompthero/openjourney",
35
+ "dreamlike-art/dreamlike-photoreal-2.0",
36
  "dreamlike-art/dreamlike-diffusion-1.0",
37
+ "sd-dreambooth-library/mr-potato-head",
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,