Update Dockerfile

Dockerfile

@@ -5,15 +5,18 @@ FROM ollama/ollama
 RUN apt-get update && apt-get install -y python3 python3-pip
 RUN pip3 install fastapi uvicorn requests
 
-# Copy a script to pull the model at container startup
+# Create a directory for persistent storage (Hugging Face Spaces uses /data)
+RUN mkdir -p /data/.ollama && ln -s /data/.ollama /root/.ollama
+
+# Copy the entrypoint script
 COPY entrypoint.sh /entrypoint.sh
 RUN chmod +x /entrypoint.sh
 
 # Copy the FastAPI app
 COPY app.py /app.py
 
-# Expose the port for the FastAPI server
+# Expose the port for the FastAPI server (Hugging Face Spaces uses 8000)
 EXPOSE 8000
 
-# Run the entrypoint script and start the FastAPI server
-CMD ["/entrypoint.sh"]
+# Run the entrypoint script using a shell
+CMD ["/bin/bash", "/entrypoint.sh"]