SYS-0000

.space.yaml

@@ -0,0 +1 @@
+sdk: docker

.streamlit/config.toml

@@ -0,0 +1,2 @@
+[browser]
+gatherUsageStats = false

Dockerfile

@@ -1,5 +1,7 @@
 FROM python:3.9-slim
 
+USER root
+
 WORKDIR /app
 
 RUN apt-get update && apt-get install -y \
@@ -15,6 +17,8 @@ COPY src/ ./src/
 
 RUN pip3 install -r requirements.txt
 
+ENV STREAMLIT_BROWSER_GATHERUSAGESTATS false
+
 EXPOSE 8501
 
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health

src/model.py

@@ -6,9 +6,11 @@ from PIL import Image
 from fastapi import FastAPI
 from typing import List
 import segmentation
+import os
+
 
 device = "cpu"
-model, preprocess = clip.load("ViT-B/32", device=device, download_root="./clip_cache")
+model, preprocess = clip.load("ViT-B/32", device=device, download_root=os.getcwd())
 
 def get_image_features(image: Image.Image) -> np.ndarray:
     """Extract CLIP features from an image."""

src/streamlit_app.py

@@ -2,6 +2,12 @@ import streamlit as st
 from PIL import Image
 import model
 
+import asyncio
+try:
+    asyncio.get_running_loop()
+except RuntimeError:
+    asyncio.set_event_loop(asyncio.new_event_loop())
+
 
 image_url_input = st.text_input("Enter the image URL:")
 k_value_input = st.number_input("Enter k_value:", min_value=1, value=5)