Update diff.py
Browse files
diff.py
CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
|
|
2 |
class SendIt:
|
3 |
def __init__(self):
|
4 |
|
5 |
-
models =[
|
6 |
"",
|
7 |
"CompVis/stable-diffusion-v1-4",
|
8 |
"runwayml/stable-diffusion-v1-5",
|
@@ -27,7 +27,7 @@ class SendIt:
|
|
27 |
|
28 |
]
|
29 |
|
30 |
-
models2=[
|
31 |
gr.Interface.load(f"models/{models[1]}",live=True,preprocess=True),
|
32 |
gr.Interface.load(f"models/{models[2]}",live=True,preprocess=True),
|
33 |
gr.Interface.load(f"models/{models[3]}",live=True,preprocess=True),
|
|
|
2 |
class SendIt:
|
3 |
def __init__(self):
|
4 |
|
5 |
+
self.models =[
|
6 |
"",
|
7 |
"CompVis/stable-diffusion-v1-4",
|
8 |
"runwayml/stable-diffusion-v1-5",
|
|
|
27 |
|
28 |
]
|
29 |
|
30 |
+
self.models2=[
|
31 |
gr.Interface.load(f"models/{models[1]}",live=True,preprocess=True),
|
32 |
gr.Interface.load(f"models/{models[2]}",live=True,preprocess=True),
|
33 |
gr.Interface.load(f"models/{models[3]}",live=True,preprocess=True),
|