Add text-to-image model app and dependencies

app.py

@@ -0,0 +1,17 @@
+import gradio as gr
+from diffusers import StableDiffusionPipeline
+import torch
+
+# Load your LoRA fine-tuned model
+model_id = "signupting/blak-lora"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe.to("cuda")  # Use GPU if available
+
+# Function to generate images
+def generate_image(prompt):
+    image = pipe(prompt).images[0]
+    return image
+
+# Gradio Interface
+demo = gr.Interface(fn=generate_image, inputs="text", outputs="image", title="Text-to-Image Generator")
+demo.launch()

requirements.txt

@@ -0,0 +1,4 @@
+torch
+gradio
+diffusers
+accelerate