xfh commited on
Commit
42da032
·
1 Parent(s): 24d2459

add .gitignore and fix weight file

Browse files
Files changed (2) hide show
  1. .gitignore +3 -0
  2. app.py +1 -1
.gitignore ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ .idea/*
2
+ __pycache__/*
3
+ flagged/*
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from stable_diffusion import Text2img, Args
2
  import gradio as gr
3
- args = Args("", 5, None, 7.5, 512, 512, 443, "cpu", "/tmp/mdjrny-v4.pt")
4
  model = Text2img.instance(args)
5
  def text2img_output(phrase):
6
  return model(phrase)
 
1
  from stable_diffusion import Text2img, Args
2
  import gradio as gr
3
+ args = Args("", 5, None, 7.5, 512, 512, 443, "cpu", "./mdjrny-v4.pt")
4
  model = Text2img.instance(args)
5
  def text2img_output(phrase):
6
  return model(phrase)