Changes, 1 worker on server and remove clip usage from server

app.py

@@ -11,7 +11,7 @@ import requests
 
 
 def start_server():
-    os.system("uvicorn server:app --port 8080 --host 0.0.0.0 --workers 2")
+    os.system("uvicorn server:app --port 8080 --host 0.0.0.0 --workers 1")
 
 
 def load_models():

server.py

@@ -12,6 +12,7 @@ from PIL import Image
 
 #import clip
 from dalle.models import Dalle
+import logging
 from dalle.utils.utils import clip_score, download
 
 print("Loading models...")
@@ -78,18 +79,20 @@ def generate(prompt):
 
 def sample(prompt):
     # Sampling
+    logging.info("starting sampling")
     images = (
         model.sampling(prompt=prompt, top_k=96, top_p=None, softmax_temperature=1.0, num_candidates=9, device=device)
         .cpu()
         .numpy()
     )
+    logging.info("sampling succeeded")
     images = np.transpose(images, (0, 2, 3, 1))
 
     # CLIP Re-ranking
-    rank = clip_score(
-        prompt=prompt, images=images, model_clip=model_clip, preprocess_clip=preprocess_clip, device=device
-    )
-    images = images[rank]
+    # rank = clip_score(
+    #     prompt=prompt, images=images, model_clip=model_clip, preprocess_clip=preprocess_clip, device=device
+    # )
+    # images = images[rank]
     
     pil_images = []
     for i in range(len(images)):