Upload 3 files

Dockerfile

@@ -0,0 +1,33 @@
+# Use an official Python runtime as a parent image
+# Using a specific version like 3.10 or 3.11 is recommended over 'latest'
+FROM python:3.10-slim
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Install system dependencies if needed (e.g., build tools) - add if required later
+# RUN apt-get update && apt-get install -y --no-install-recommends some-package && rm -rf /var/lib/apt/lists/*
+
+# Copy the requirements file into the container
+COPY requirements.txt .
+
+# Install Python dependencies
+# --no-cache-dir reduces image size
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt
+
+# Copy the rest of the application code into the container
+COPY . .
+
+# Explicitly create cache dir and set open permissions during build
+RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
+
+# Set environment variables for Hugging Face cache directories
+ENV HF_HOME=/app/.cache/huggingface
+ENV HF_DATASETS_CACHE=/app/.cache/datasets
+
+# Expose the port the app runs on
+EXPOSE 7860
+
+# Define the command to run the application
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,27 @@
+from fastapi import FastAPI, Response, HTTPException
+from datasets import load_dataset
+from io import BytesIO
+from PIL import Image
+
+app = FastAPI()
+
+dataset = load_dataset("visionLMsftw/vibe-testing-samples", split="train")
+id_to_image = {example["ex_id"]: example["image"] for example in dataset}
+
+print(id_to_image)
+
+@app.get("/image/{image_id}")
+def get_image(image_id: int):
+    if image_id not in id_to_image:
+        raise HTTPException(status_code=404, detail="Image not found")
+
+    image: Image.Image = id_to_image[image_id].convert("RGB")
+    buffer = BytesIO()
+    image.save(buffer, format="JPEG", quality=85)
+    buffer.seek(0)
+    return Response(content=buffer.read(), media_type="image/jpeg")
+
+@app.get("/ids")
+def list_ids():
+    return list(id_to_image.keys())
+

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+datasets
+pillow