Update app.py

app.py

@@ -1,25 +1,22 @@
-from stable_diffusion_tf.stable_diffusion import Text2Image
-from PIL import Image
 import gradio as gr
+import torch
+import numpy as np
 import modin.pandas as pd
+from PIL import Image
+from diffusers import DiffusionPipeline
 
-generator = Text2Image(
-    img_height=512,
-    img_width=512,
-    jit_compile=False)
-
-def txt2img(prompt, guide, steps, Temp):
-    img = generator.generate(prompt,
-    num_steps=steps,
-    unconditional_guidance_scale=guide,
-    temperature=Temp,
-    batch_size=1)
-    image=Image.fromarray(img[0])
-    return image
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney-v2", safety_checker=None)
+pipe = pipe.to(device)
 
-iface = gr.Interface(fn=txt2img, inputs=[
-    gr.Textbox(label = 'Input Text Prompt: 77 Token (Keyword) Limit'),
-    gr.Slider(1, 25, value = 10, step = 1, label = 'Guidance Scale: How close to follow Prompt'),
-    gr.Slider(20, 75, value = 25, step = 1, label = 'Number of Iterations: Anything above 50 may produce the Over Baked Effect'),
-    gr.Slider(.1, 100, value = 1, label='Temperature: Changes probability of Diffusion to Image Array, more info in community comments')], outputs = 'image',title='Stable Diffusion with Keras and TensorFlow CPU or GPU', description='Now Using Keras and TensorFlow with Stable Diffusion. This allows very complex image generation with less code footprint, and less text. Simply type in what you wish to see, adjust the sliders (optional) and click submit. For more information on Keras see https://keras.io/about/ For more informationon about Stable Diffusion or Suggestions for prompts, keywords, artists or styles see https://github.com/Maks-s/sd-akashic <br><br><b>DISCLAIMER: This Text to Image Pipeline has the potential to produce NSFW, Disturbing, or Offensive Images. I am not responsible for what images you produce or what you do with them. By using this Gradio API you consent to taking full responsibility for the images you produce and agree that you are at least 18 years of age.', article = "Code Monkey: <a href=\"https://huggingface.co/Manjushri\">Manjushri</a>")
-iface.launch(max_threads=True)
+def genie (prompt, scale, steps, seed):
+     generator = torch.Generator(device=device).manual_seed(seed)
+     images = pipe(prompt, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]
+     return images
+    
+gr.Interface(fn=genie, inputs=[gr.Textbox(label='What you want the AI to generate. 77 Token Limit.'), 
+                               gr.Slider(1, maximum=15, value=10, step=.25), 
+                               gr.Slider(1, maximum=50, value=25, step=1), 
+                               gr.Slider(minimum=1, step=1, maximum=987654321, randomize=True)], 
+             outputs='image', title="OpenJourney V2 CPU", description="OJ V2 CPU. <b>WARNING:</b> Extremely Slow. 35s/Iteration. Expect 8-16mins an image for 15-30 iterations respectively. 50 iterations takes ~28mins.", 
+             article = "Code Monkey: <a href=\"https://huggingface.co/Manjushri\">Manjushri</a>").launch(debug=True, max_threads=True)