Update server.py
Browse files
server.py
CHANGED
|
@@ -33,7 +33,7 @@ CORS(app)
|
|
| 33 |
stable_diff_pipe = None
|
| 34 |
model = None
|
| 35 |
|
| 36 |
-
def load_models(model_name="
|
| 37 |
global stable_diff_pipe, model
|
| 38 |
|
| 39 |
# Load Stable Diffusion model pipeline
|
|
@@ -79,7 +79,7 @@ def static_files(filename):
|
|
| 79 |
@app.route('/api/check-membership', methods=['POST'])
|
| 80 |
def check_membership():
|
| 81 |
try:
|
| 82 |
-
model_name = request.form.get('model', '
|
| 83 |
global stable_diff_pipe, model
|
| 84 |
if stable_diff_pipe is None or model is None:
|
| 85 |
load_models(model_name)
|
|
|
|
| 33 |
stable_diff_pipe = None
|
| 34 |
model = None
|
| 35 |
|
| 36 |
+
def load_models(model_name="openai/clip-vit-base-patch32"):
|
| 37 |
global stable_diff_pipe, model
|
| 38 |
|
| 39 |
# Load Stable Diffusion model pipeline
|
|
|
|
| 79 |
@app.route('/api/check-membership', methods=['POST'])
|
| 80 |
def check_membership():
|
| 81 |
try:
|
| 82 |
+
model_name = request.form.get('model', 'openai/clip-vit-base-patch32')
|
| 83 |
global stable_diff_pipe, model
|
| 84 |
if stable_diff_pipe is None or model is None:
|
| 85 |
load_models(model_name)
|