JoPmt commited on
Commit
a08f7b3
1 Parent(s): b3538b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -27
app.py CHANGED
@@ -28,30 +28,17 @@ models =[
28
  "CompVis/stable-diffusion-v1-4",
29
  "stabilityai/stable-diffusion-2-1",
30
  "stablediffusionapi/edge-of-realism",
31
- "wavymulder/lomo-diffusion",
32
  "sd-dreambooth-library/fashion",
33
  "DucHaiten/DucHaitenDreamWorld",
34
  "kandinsky-community/kandinsky-2-1",
35
- "MirageML/lowpoly-cyberpunk",
36
  "plasmo/woolitize-768sd1-5",
37
- "plasmo/food-crit",
38
- "johnslegers/epic-diffusion-v1.1",
39
- "Fictiverse/ElRisitas",
40
- "robotjung/SemiRealMix",
41
- "herpritts/FFXIV-Style",
42
- "RayHell/popupBook-diffusion",
43
- "deadman44/SD_Photoreal_Merged_Models",
44
- "johnslegers/epic-diffusion",
45
- "tilake/China-Chic-illustration",
46
  "wavymulder/modelshoot",
47
- "prompthero/openjourney-lora",
48
  "Fictiverse/Stable_Diffusion_VoxelArt_Model",
49
  "darkstorm2150/Protogen_v2.2_Official_Release",
50
  "hassanblend/HassanBlend1.5.1.2",
51
  "hassanblend/hassanblend1.4",
52
  "nitrosocke/redshift-diffusion",
53
  "prompthero/openjourney-v2",
54
- "nitrosocke/Arcane-Diffusion",
55
  "Lykon/DreamShaper",
56
  "nitrosocke/mo-di-diffusion",
57
  "dreamlike-art/dreamlike-diffusion-1.0",
@@ -59,21 +46,8 @@ models =[
59
  "digiplay/RealismEngine_v1",
60
  "digiplay/AIGEN_v1.4_diffusers",
61
  "stablediffusionapi/dreamshaper-v6",
62
- "JackAnon/GorynichMix",
63
- "p1atdev/liminal-space-diffusion",
64
- "nadanainone/gigaschizonegs",
65
- "darkVOYAGE/dvMJv4",
66
- "digiplay/GhostMix",
67
- "ThePioneer/MISA",
68
  "TheLastBen/froggy-style-v21-768",
69
- "FloydianSound/Nixeu_Diffusion_v1-5",
70
- "kakaobrain/karlo-v1-alpha-image-variations",
71
  "digiplay/PotoPhotoRealism_v1",
72
- "ConsistentFactor/Aurora-By_Consistent_Factor",
73
- "rim0/quadruped_mechas",
74
- "Akumetsu971/SD_Samurai_Anime_Model",
75
- "sd-dreambooth-library/original-character-cyclps",
76
- "AIArtsChannel/steampunk-diffusion",
77
  ]
78
 
79
  controlnet = accelerator.prepare(ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.float32))
@@ -173,6 +147,6 @@ def plex(fpath, text, neg_prompt, modil, one, two, three, four, five):
173
  export_to_video(canframes, canvid, fps=fps)
174
  return video, canvid, orvid
175
 
176
- iface = gr.Interface(fn=plex, inputs=[gr.File(label="Your video",interactive=True),gr.Textbox(label="prompt"),gr.Textbox(label="neg prompt"),gr.Dropdown(choices=models, label="some sd models", value=models[0], type="value"), gr.Slider(label="num inference steps", minimum=1, step=1, maximum=10, value=4), gr.Slider(label="prompt strength", minimum=0.01, step=0.01, maximum=20.00, value=5.00), gr.Slider(label="controlnet scale", minimum=0.01, step=0.01, maximum=0.99, value=0.80), gr.Slider(label="Guidance scale", minimum=0.01, step=0.01, maximum=10.00, value=2.00), gr.Slider(label="Manual seed", minimum=0, step=32, maximum=4836928, value=0)], outputs=[gr.Video(label="final"), gr.Video(label="canny vid"), gr.Video(label="orig")],description="Running on cpu, very slow! by JoPmt.")
177
  iface.queue(max_size=1,api_open=False)
178
  iface.launch(max_threads=1)
 
28
  "CompVis/stable-diffusion-v1-4",
29
  "stabilityai/stable-diffusion-2-1",
30
  "stablediffusionapi/edge-of-realism",
 
31
  "sd-dreambooth-library/fashion",
32
  "DucHaiten/DucHaitenDreamWorld",
33
  "kandinsky-community/kandinsky-2-1",
 
34
  "plasmo/woolitize-768sd1-5",
 
 
 
 
 
 
 
 
 
35
  "wavymulder/modelshoot",
 
36
  "Fictiverse/Stable_Diffusion_VoxelArt_Model",
37
  "darkstorm2150/Protogen_v2.2_Official_Release",
38
  "hassanblend/HassanBlend1.5.1.2",
39
  "hassanblend/hassanblend1.4",
40
  "nitrosocke/redshift-diffusion",
41
  "prompthero/openjourney-v2",
 
42
  "Lykon/DreamShaper",
43
  "nitrosocke/mo-di-diffusion",
44
  "dreamlike-art/dreamlike-diffusion-1.0",
 
46
  "digiplay/RealismEngine_v1",
47
  "digiplay/AIGEN_v1.4_diffusers",
48
  "stablediffusionapi/dreamshaper-v6",
 
 
 
 
 
 
49
  "TheLastBen/froggy-style-v21-768",
 
 
50
  "digiplay/PotoPhotoRealism_v1",
 
 
 
 
 
51
  ]
52
 
53
  controlnet = accelerator.prepare(ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.float32))
 
147
  export_to_video(canframes, canvid, fps=fps)
148
  return video, canvid, orvid
149
 
150
+ iface = gr.Interface(fn=plex, inputs=[gr.File(label="Your video",interactive=True, file_types=".mp4"),gr.Textbox(label="prompt"),gr.Textbox(label="neg prompt"),gr.Dropdown(choices=models, label="Models", value=models[0], type="value"), gr.Slider(label="num inference steps", minimum=1, step=1, maximum=10, value=4), gr.Slider(label="Strength", minimum=0.01, step=0.01, maximum=20.00, value=5.00), gr.Slider(label="controlnet scale", minimum=0.01, step=0.01, maximum=0.99, value=0.80), gr.Slider(label="Guidance scale", minimum=0.01, step=0.01, maximum=10.00, value=2.00), gr.Slider(label="Manual seed", minimum=0, step=32, maximum=4836928, value=0)], outputs=[gr.Video(label="final"), gr.Video(label="canny vid"), gr.Video(label="orig")],description="Running on cpu, very slow! by JoPmt.")
151
  iface.queue(max_size=1,api_open=False)
152
  iface.launch(max_threads=1)