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()
|