Update server.py
Browse files
server.py
CHANGED
|
@@ -35,6 +35,7 @@ CORS(app)
|
|
| 35 |
stable_diff_pipe = None
|
| 36 |
model = None
|
| 37 |
|
|
|
|
| 38 |
db = TinyDB('/data/artwork_results.json')
|
| 39 |
|
| 40 |
def load_models(model_name="rupeshs/LCM-runwayml-stable-diffusion-v1-5"):
|
|
|
|
| 35 |
stable_diff_pipe = None
|
| 36 |
model = None
|
| 37 |
|
| 38 |
+
os.makedirs('/data', exist_ok=True)
|
| 39 |
db = TinyDB('/data/artwork_results.json')
|
| 40 |
|
| 41 |
def load_models(model_name="rupeshs/LCM-runwayml-stable-diffusion-v1-5"):
|