feat: add GitHub Actions workflow and Docker setup

- Add GitHub Actions workflow for syncing to Hugging Face hub
- Create Dockerfile with PyTorch and CUDA support
- Implement Streamlit app for text-to-image generation
- Add requirements.txt for dependencies

.github/workflows/main.yml

@@ -0,0 +1,23 @@
+name: Sync to Hugging Face hub
+on:
+  push:
+    branches: [main]
+
+  # to run this workflow manually from the Actions tab
+  workflow_dispatch:
+
+jobs:
+  sync-to-hub:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v2
+        with:
+          fetch-depth: 0
+      - name: Add remote 
+        env:
+          HF: ${{ secrets.HF }}
+        run: git remote add space https://argnctu:[email protected]/spaces/argnctu/text-to-image
+      - name: Push to hub
+        env:
+          HF: ${{ secrets.HF }}
+        run: git push --force https://argnctu:[email protected]/spaces/argnctu/text-to-image main

Dockerfile

@@ -0,0 +1,35 @@
+# Base image with PyTorch and CUDA support
+FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu20.04
+
+# Set environment variables
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
+
+# Install essential packages
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3.8 \
+    python3-pip \
+    python3-venv \
+    git \
+    wget \
+    libgl1-mesa-glx \
+    && apt-get clean && rm -rf /var/lib/apt/lists/*
+
+# Set Python aliases
+RUN ln -s /usr/bin/python3.8 /usr/bin/python && \
+    ln -s /usr/bin/pip3 /usr/bin/pip
+
+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
+EXPOSE 7860
+
+CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]

app.py

@@ -0,0 +1,32 @@
+import streamlit as st
+from PIL import Image
+from flux import FLUXModel  # Assuming FLUX.1-dev provides this
+
+# Initialize the FLUX model (load only once)
+@st.cache_resource
+def load_model():
+    return FLUXModel.from_pretrained("flux-1-dev")
+
+model = load_model()
+
+# Streamlit GUI
+st.title("Text-to-Image Generator with FLUX.1-dev")
+st.subheader("Enter a prompt to generate stunning images!")
+
+# Text input
+prompt = st.text_input("Enter your prompt", "")
+
+# Generate image when button is clicked
+if st.button("Generate Image"):
+    if prompt:
+        with st.spinner("Generating image... Please wait."):
+            try:
+                image = model.generate_image(prompt)  # Generate image
+                st.image(image, caption="Generated Image", use_column_width=True)
+            except Exception as e:
+                st.error(f"Error generating image: {e}")
+    else:
+        st.warning("Please enter a prompt!")
+
+# Footer
+st.write("Powered by FLUX.1-dev and Streamlit")

requirements.txt

@@ -0,0 +1,9 @@
+torch>=2.0.0
+transformers>=4.28.0
+diffusers>=0.15.0
+accelerate>=0.19.0
+pillow
+numpy
+scipy
+flux-1-dev
+streamlit