Manjushri commited on
Commit
4b5437e
1 Parent(s): 934504b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,7 +6,7 @@ from PIL import Image
6
  from diffusers import DiffusionPipeline
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
- pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney-v2", torch_dtype=torch.float16, safety_checker=None)
10
  pipe = pipe.to(device)
11
 
12
  def genie (prompt, scale, steps, Seed):
@@ -19,6 +19,6 @@ gr.Interface(fn=genie, inputs=[gr.Textbox(label='What you want the AI to generat
19
  gr.Slider(1, maximum=100, value=50, step=1, label='Number of Iterations: 50 is typically fine.'),
20
  gr.Slider(minimum=1, step=1, maximum=987654321, randomize=True, interactive=True)],
21
  outputs=gr.Image(label='512x512 Generated Image'),
22
- title="OpenJourney V2 GPU",
23
- description="OJ V2 GPU. Ultra Fast, now running on a T4",
24
  article = "Code Monkey: <a href=\"https://huggingface.co/Manjushri\">Manjushri</a>").launch(debug=True, max_threads=True)
 
6
  from diffusers import DiffusionPipeline
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
+ pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney-v4", torch_dtype=torch.float16, safety_checker=None)
10
  pipe = pipe.to(device)
11
 
12
  def genie (prompt, scale, steps, Seed):
 
19
  gr.Slider(1, maximum=100, value=50, step=1, label='Number of Iterations: 50 is typically fine.'),
20
  gr.Slider(minimum=1, step=1, maximum=987654321, randomize=True, interactive=True)],
21
  outputs=gr.Image(label='512x512 Generated Image'),
22
+ title="OpenJourney V4 GPU",
23
+ description="OJ V4 GPU. Ultra Fast, now running on a T4",
24
  article = "Code Monkey: <a href=\"https://huggingface.co/Manjushri\">Manjushri</a>").launch(debug=True, max_threads=True)