Muhammadreza commited on
Commit
0c1adc7
·
verified ·
1 Parent(s): f04735b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -3,7 +3,9 @@ import gradio as gr
3
  import torch
4
  import modin.pandas as pd
5
  import numpy as np
6
- from diffusers import DiffusionPipeline, DPMSolverSinglestepScheduler
 
 
7
 
8
  pipe = DiffusionPipeline.from_pretrained("mann-e/Mann-E_Dreams", torch_dtype=torch.float16).to("cuda")
9
  pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config, use_karras_sigmas=True)
@@ -21,7 +23,7 @@ gr.Interface(fn=genie, inputs=[gr.Textbox(label='What you want the AI to generat
21
  gr.Slider(576, maximum=1280, value=768, step=16, label='Width (can go up to 1280, but for square images maximum is 1024x1024)'),
22
  gr.Slider(576, maximum=1280, value=768, step=16, label='Height (can go up to 1280, but for square images maximum is 1024x1024)'),
23
  gr.Slider(1, maximum=8, value=6, step=1, label='Number of Iterations'),
24
- gr.Slider(minimum=0, step=1, maximum=999999999999999999, randomize=True),
25
  ],
26
  outputs='image',
27
  title="Mann-E Dreams",
 
3
  import torch
4
  import modin.pandas as pd
5
  import numpy as np
6
+ from diffusers import DiffusionPipeline, DPMSolverSinglestepScheduler, ControlNetModel, StableDiffusionXLControlNetPipeline, AutoencoderKL
7
+ import requests
8
+ import cv2
9
 
10
  pipe = DiffusionPipeline.from_pretrained("mann-e/Mann-E_Dreams", torch_dtype=torch.float16).to("cuda")
11
  pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config, use_karras_sigmas=True)
 
23
  gr.Slider(576, maximum=1280, value=768, step=16, label='Width (can go up to 1280, but for square images maximum is 1024x1024)'),
24
  gr.Slider(576, maximum=1280, value=768, step=16, label='Height (can go up to 1280, but for square images maximum is 1024x1024)'),
25
  gr.Slider(1, maximum=8, value=6, step=1, label='Number of Iterations'),
26
+ gr.Slider(minimum=0, step=1, maximum=999999999999999999, randomize=True, label="Seed"),
27
  ],
28
  outputs='image',
29
  title="Mann-E Dreams",