Updated requirements and modified script

README.md

@@ -1,13 +1,18 @@
 ---
-title: Stable Diffusion 2 1
-emoji: 🖼
-colorFrom: purple
-colorTo: red
+title: "AI Image Generator"
+emoji: 🎨
+colorFrom: blue
+colorTo: purple
 sdk: gradio
-sdk_version: 5.0.1
-app_file: app.py
-pinned: false
-short_description: Public text-to-image AI model
+python_version: 3.10
+sdk_version: 4.0
+suggested_hardware: "cpu-basic"
+app_file: "app.py"
+models:
+  - prithivMLmods/SD3.5-Turbo-Realism-2.0-LoRA
+  - Purz/face-projection
+tags:
+  - stable-diffusion
+  - text-to-image
+pinned: true
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,30 +1,51 @@
 import gradio as gr
+import threading
+import os
 import torch
-from diffusers import StableDiffusionPipeline
 
-# Force using CPU
-device = "cpu"
+os.environ["OMP_NUM_THREADS"] = str(os.cpu_count())
+torch.set_num_threads(os.cpu_count())
 
-# Load the model on CPU
-pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype=torch.float32)
-pipe.to(device)
+model1 = gr.load("models/prithivMLmods/SD3.5-Turbo-Realism-2.0-LoRA")
+model2 = gr.load("models/Purz/face-projection")
 
-# Function to generate images
-def generate_image(prompt):
-    with torch.no_grad():  # Disable gradient calculation
-        image = pipe(prompt=prompt).images[0]  # Generate image
-    return image
+stop_event = threading.Event()
 
-# Gradio UI
-with gr.Blocks() as demo:
-    gr.Markdown("# Stable Diffusion 2.1 - Image Generator")
+def generate_image(text, selected_model):
+    stop_event.clear()
+
+    if selected_model == "Model 1 (Turbo Realism)":
+        model = model1
+    elif selected_model == "Model 2 (Face Projection)":
+        model = model2
+    else:
+        return "Invalid model selection."
+
+    if stop_event.is_set():
+        return "Image generation stopped by user."
+
+    return model(text)
+
+def stop_generation():
+    """Stops the ongoing image generation by setting the stop_event flag."""
+    stop_event.set()
+    return "Generation stopped."
+
+with gr.Blocks() as interface:
+    text_input = gr.Textbox(label="Type your prompt:", placeholder="Enter your imagination...")
+    model_selector = gr.Radio(
+        ["Model 1 (Turbo Realism)", "Model 2 (Face Projection)"],
+        label="Select Model",
+        value="Model 1 (Turbo Realism)"
+    )
     
     with gr.Row():
-        prompt = gr.Textbox(label="Enter Prompt", placeholder="Describe the image you want to generate")
-        btn = gr.Button("Generate")
-
+        generate_button = gr.Button("Generate Image 🎨")
+        stop_button = gr.Button("Stop Image Generation")
+    
     output = gr.Image(label="Generated Image")
     
-    btn.click(fn=generate_image, inputs=prompt, outputs=output)
+    generate_button.click(generate_image, inputs=[text_input, model_selector], outputs=output)
+    stop_button.click(stop_generation, inputs=[], outputs=output)
 
-demo.launch(share=True)  # Optional: Share link
+interface.launch()

requirements.txt

@@ -1,5 +1,81 @@
-gradio
-diffusers
-torch
-torchvision
-transformers
+aiofiles==23.2.1
+altair==5.5.0
+annotated-types==0.7.0
+anyio==4.9.0
+attrs==25.3.0
+certifi==2025.1.31
+charset-normalizer==3.4.1
+click==8.1.8
+colorama==0.4.6
+contourpy==1.3.1
+cycler==0.12.1
+diffusers==0.32.2
+exceptiongroup==1.2.2
+fastapi==0.115.12
+ffmpy==0.5.0
+filelock==3.18.0
+fonttools==4.56.0
+fsspec==2025.3.0
+gradio==3.50.2
+gradio_client==0.6.1
+groovy==0.1.2
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.28.1
+huggingface-hub==0.29.3
+idna==3.10
+importlib_metadata==8.6.1
+importlib_resources==6.5.2
+Jinja2==3.1.6
+jsonschema==4.23.0
+jsonschema-specifications==2024.10.1
+kiwisolver==1.4.8
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.10.1
+mdurl==0.1.2
+mpmath==1.3.0
+narwhals==1.32.0
+networkx==3.4.2
+numpy==1.26.4
+orjson==3.10.16
+packaging==24.2
+pandas==2.2.3
+pillow==10.4.0
+pydantic==2.11.0
+pydantic_core==2.33.0
+pydub==0.25.1
+Pygments==2.19.1
+pyparsing==3.2.3
+python-dateutil==2.9.0.post0
+python-multipart==0.0.20
+pytz==2025.2
+PyYAML==6.0.2
+referencing==0.36.2
+regex==2024.11.6
+requests==2.32.3
+rich==13.9.4
+rpds-py==0.24.0
+ruff==0.11.2
+safehttpx==0.1.6
+safetensors==0.5.3
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.17.0
+sniffio==1.3.1
+starlette==0.46.1
+sympy==1.13.1
+tokenizers==0.21.1
+tomlkit==0.13.2
+torch==2.6.0
+torchvision==0.21.0
+tqdm==4.67.1
+transformers==4.50.2
+typer==0.15.2
+typing-inspection==0.4.0
+typing_extensions==4.13.0
+tzdata==2025.2
+urllib3==2.3.0
+uvicorn==0.34.0
+websockets==11.0.3
+zipp==3.21.0