Pls work

.gitignore

@@ -1,3 +1,4 @@
 __pycache__
 node_modules
-poetry.lock
+poetry.lock
+cache

Dockerfile

@@ -4,13 +4,9 @@ WORKDIR /api
 
 COPY ./ /api/
 
-RUN pip install --upgrade pip
-RUN pip install -r requirements.txt
-RUN poetry install --no-root
-
 RUN mkdir /api/cache
 RUN chmod a+rwx /api/cache
 
 EXPOSE 7860
 
-CMD ["python", "main.py"]
+CMD ["bash", "start.sh"]

main.py

@@ -20,8 +20,8 @@ text = "a cat sitting on a table"
 # Tokenize text and get features
 inputs = clip_processor(text, return_tensors="pt", padding=True)
 
-with torch.no_grad():
-  image_features = clip_model.get_image_features(**inputs)
+# Generate image features
+image_features = clip_model.get_image_features(pixel_values=inputs.pixel_values)
 
 # Generate image from features
 generated_image = clip_model.generate_images(image_features)

start.sh

@@ -0,0 +1,5 @@
+pip install --upgrade pip
+pip install -r requirements.txt
+poetry install --no-root
+
+python main.py