changes

app.py

@@ -0,0 +1,33 @@
+import gradio as gr
+import numpy as np
+from PIL import Image
+import torch
+import safetensors.torch as st
+from diffusers import DiffusionPipeline
+
+model_id = "./ckpts/snckrsgen.safetensors"
+pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, use_safe_tensors=True)
+pipe.to("cuda")
+
+# Function to generate an image from text using diffusion
+def generate_image(prompt):
+    
+    images = pipe(prompt=prompt)
+    return images[0]
+
+# Gradio interface
+inputs = gr.inputs.Textbox(lines=5, label="Enter text to generate image:")
+outputs = gr.outputs.Image(label="Generated Image")
+
+title = "ShoeGen: Generate an Image of a Shoe"
+description = "Enter a text description of a shoe to generate an image of the shoe."
+examples = [["A red shoe with white laces and black sole."], ["A blue sneaker with a white stripe."], ["A brown boot with a buckle."]]
+
+gr.Interface(
+    fn=generate_image,
+    inputs=inputs,
+    outputs=outputs,
+    title=title,
+    description=description,
+    examples=examples
+).launch()

ckpts/snckrsgen.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e80bf5f4ded84793d74c9939b0fc1a09b76af31bafe2ac3190c21c9be5eb6965
+size 151112168

requirements.txt

@@ -0,0 +1,13 @@
+torch
+torchvision
+diffusers
+gradio
+numpy
+PIL
+safetensors
+scipy
+ftfy
+psutil
+accelerate==0.12.0
+transformers
+https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.15/xformers-0.0.15.dev0+4c06c79.d20221205-cp38-cp38-linux_x86_64.whl