Revert to Gradio SDK, simplify requirements.txt, add runtime dependency installation, remove Dockerfile

Dockerfile

@@ -1,14 +0,0 @@
-FROM pytorch/pytorch:2.4.1-cuda12.1-cudnn9-runtime
-RUN pip install --no-cache-dir \
-    accelerate==0.33.0 \
-    diffusers==0.34.0 \
-    gradio==4.44.0 \
-    spaces>=0.28.3 \
-    numpy==1.26.4 \
-    pillow==10.4.0 \
-    transformers==4.44.2 \
-    safetensors==0.4.5 \
-    tqdm==4.66.5
-COPY . /app
-WORKDIR /app
-CMD ["python", "app.py"]

README.md

@@ -3,7 +3,7 @@ title: SD Imagic
 emoji: 🐢
 colorFrom: red
 colorTo: yellow
-sdk: docker
+sdk: gradio
 app_file: app.py
 pinned: false
 ---

app.py

@@ -1,9 +1,32 @@
+import os
+import subprocess
+import sys
+
+# Install dependencies at runtime to handle Dev Mode resets
+def install_packages():
+    packages = [
+        "accelerate",
+        "diffusers",
+        "gradio",
+        "numpy",
+        "pillow",
+        "torch",
+        "transformers",
+        "safetensors",
+        "tqdm"
+    ]
+    for package in packages:
+        try:
+            subprocess.check_call([sys.executable, "-m", "pip", "install", package])
+        except subprocess.CalledProcessError as e:
+            print(f"Failed to install {package}: {e}")
+
+install_packages()
+
 import gradio as gr
 import numpy as np
-import os
 import torch
 import random
-import subprocess
 from imagic_stable_diffusion import ImagicStableDiffusionPipeline
 from PIL import Image
 

requirements.txt

@@ -1,10 +1,9 @@
-accelerate==0.33.0
-diffusers==0.34.0
-gradio==4.44.0
-spaces>=0.28.3
-numpy==1.26.4
-pillow==10.4.0
-torch==2.4.1
-transformers==4.44.2
-safetensors==0.4.5
-tqdm==4.66.5
+accelerate
+diffusers
+gradio
+numpy
+pillow
+torch
+transformers
+safetensors
+tqdm