RoyalEagle
commited on
Commit
•
46ccdda
1
Parent(s):
62d725a
Update app.py
Browse files
app.py
CHANGED
@@ -9,12 +9,15 @@ os.system("git clone https://github.com/CompVis/taming-transformers.git")
|
|
9 |
# !pip install taming-transformers
|
10 |
#clone https://github.com/CompVis/taming-transformers.git
|
11 |
os.system("git clone https://github.com/dribnet/pixray")
|
|
|
|
|
12 |
import gradio as gr
|
13 |
import torch
|
14 |
sys.path.append("pixray")
|
15 |
import pixray
|
16 |
|
17 |
|
|
|
18 |
# Define the main function
|
19 |
def generate(prompt, quality, aspect):
|
20 |
torch.cuda.empty_cache()
|
|
|
9 |
# !pip install taming-transformers
|
10 |
#clone https://github.com/CompVis/taming-transformers.git
|
11 |
os.system("git clone https://github.com/dribnet/pixray")
|
12 |
+
os.mkdir("steps")
|
13 |
+
os.mkdir("models")
|
14 |
import gradio as gr
|
15 |
import torch
|
16 |
sys.path.append("pixray")
|
17 |
import pixray
|
18 |
|
19 |
|
20 |
+
|
21 |
# Define the main function
|
22 |
def generate(prompt, quality, aspect):
|
23 |
torch.cuda.empty_cache()
|