Upload 3 files

Dockerfile

@@ -0,0 +1,13 @@
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,61 @@
+from fastapi import FastAPI
+from fastapi.responses import StreamingResponse
+import urllib.request
+import io
+
+app = FastAPI()
+
+@app.get("/download-image")
+async def download_image(path: str):
+    BASE_URL = "https://kwai-kolors-kolors-virtual-try-on.hf.space/file="
+    
+    try:
+        # Download image
+        with urllib.request.urlopen(BASE_URL+path) as response:
+            image_data = response.read()
+        
+        # Create a BytesIO object for streaming
+        image_stream = io.BytesIO(image_data)
+        
+        # Return StreamingResponse with appropriate media type
+        return StreamingResponse(
+            image_stream,
+            media_type="image/webp",
+            headers={
+                "Content-Disposition": "inline; filename=image.webp"
+            }
+        )
+    except Exception as e:
+        return {"error": str(e)}
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+'''import urllib.request
+
+url = 'https://kwai-kolors-kolors-virtual-try-on.hf.space/file=/tmp/gradio/e3f14f9d34b839a2b2a338b8e4a1827f6a1902663ddb031a8bb2e296b1bdcbf7/image.webp'
+
+def download_image_and_log_bytes(url, save_as):
+    with urllib.request.urlopen(url) as response:
+        image_data = response.read()
+        print(f"[LOG] Image byte length: {len(image_data)}")
+        print(f"[LOG] First 100 bytes: {image_data[:100]}")
+        with open(save_as, 'wb') as f:
+            f.write(image_data)
+
+save_as = 'imager.jpg'
+download_image_and_log_bytes(url, save_as)'''

requirements.txt

@@ -0,0 +1,11 @@
+fastapi
+uvicorn[standard]
+requests
+python-multipart
+gradio_client
+pydantic
+httpx
+sse-starlette
+aiohttp
+urllib3
+wget