Update server.py

server.py

@@ -33,7 +33,7 @@ CORS(app)
 stable_diff_pipe = None
 model = None
 
-def load_models(model_name="openai/clip-vit-base-patch32"):
+def load_models(model_name="CompVis/stable-diffusion-v1-4"):
     global stable_diff_pipe, model
     
     # Load Stable Diffusion model pipeline
@@ -79,7 +79,7 @@ def static_files(filename):
 @app.route('/api/check-membership', methods=['POST'])
 def check_membership():
     try:
-        model_name = request.form.get('model', 'openai/clip-vit-base-patch32')
+        model_name = request.form.get('model', 'CompVis/stable-diffusion-v1-4')
         global stable_diff_pipe, model
         if stable_diff_pipe is None or model is None:
             load_models(model_name)