changes

app.py

@@ -1,13 +1,13 @@
 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)
+model_id = "Apocalypse-19/shoe-generator"
+pipe = DiffusionPipeline.from_pretrained(model_id)
+
 pipe.to("cuda")
+pipe.enable_xformers_memory_efficient_attention()
+
 
 # Function to generate an image from text using diffusion
 def generate_image(prompt):

requirements.txt

@@ -1,13 +1,6 @@
 torch
-torchvision
-diffusers
 gradio
-numpy
-PIL
-safetensors
-scipy
-ftfy
-psutil
-accelerate==0.12.0
-transformers
-xformers
+diffusers==0.26.3
+transformers==4.38.1
+accelerate==0.27.2
+omegaconf