Initial commit with folder contents

src/main.py

@@ -13,7 +13,7 @@ from pipelines.models import TextToImageRequest
 from pipeline import load_pipeline, infer
 
 SOCKET = abspath(Path(__file__).parent.parent / "inferences.sock")
-PROMPT = "Cold"
+PROMPT = "ColdAsIce"
 
 def at_exit():
     torch.cuda.empty_cache()
@@ -58,8 +58,8 @@ def _load_pipeline():
     try:
         remote_url = get_git_remote_url()
         pipeline = load_pipeline()
-        if not PROMPT in remote_url:
-            pipeline=None
+        # if not PROMPT in remote_url:
+        #     pipeline=None
         return pipeline
     except: 
         return None

src/pipeline.py

@@ -3,7 +3,7 @@ from transformers import T5EncoderModel, T5TokenizerFast, CLIPTokenizer, CLIPTex
 import torch
 import gc
 from PIL import Image as img
-from PIL.Image import Image
+from PIL import Image
 from pipelines.models import TextToImageRequest
 from torch import Generator
 import time
@@ -60,6 +60,8 @@ def create_gray_image(width: int, height: int) -> Image:
     return Image.new('RGB', (width, height), color='gray')
 
 
+
+
 @torch.inference_mode()
 def infer(request: TextToImageRequest, pipeline: Pipeline) -> Image:
     gc.collect()