Update Dockerfile

Dockerfile

@@ -8,9 +8,9 @@ RUN pip3 install fastapi uvicorn requests
 # 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 startup script for model pulling and FastAPI
+COPY startup.sh /startup.sh
+RUN chmod +x /startup.sh
 
 # Copy the FastAPI app
 COPY app.py /app.py
@@ -18,5 +18,8 @@ COPY app.py /app.py
 # Expose the port for the FastAPI server (Hugging Face Spaces uses 8000)
 EXPOSE 8000
 
-# Use the default Ollama entrypoint and pass the custom script as a command
-CMD ["/entrypoint.sh"]
+# Run ollama serve as the main command
+CMD ["serve"]
+
+# Start the startup script in the background during container initialization
+RUN echo "/startup.sh &" >> /etc/profile.d/startup.sh