Omnibus commited on
Commit
232bf6d
1 Parent(s): 4714325

Update diff.py

Browse files
Files changed (1) hide show
  1. diff.py +2 -2
diff.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  class SendIt:
3
  def __init__(self):
4
 
5
- models =[
6
  "",
7
  "CompVis/stable-diffusion-v1-4",
8
  "runwayml/stable-diffusion-v1-5",
@@ -27,7 +27,7 @@ class SendIt:
27
 
28
  ]
29
 
30
- models2=[
31
  gr.Interface.load(f"models/{models[1]}",live=True,preprocess=True),
32
  gr.Interface.load(f"models/{models[2]}",live=True,preprocess=True),
33
  gr.Interface.load(f"models/{models[3]}",live=True,preprocess=True),
 
2
  class SendIt:
3
  def __init__(self):
4
 
5
+ self.models =[
6
  "",
7
  "CompVis/stable-diffusion-v1-4",
8
  "runwayml/stable-diffusion-v1-5",
 
27
 
28
  ]
29
 
30
+ self.models2=[
31
  gr.Interface.load(f"models/{models[1]}",live=True,preprocess=True),
32
  gr.Interface.load(f"models/{models[2]}",live=True,preprocess=True),
33
  gr.Interface.load(f"models/{models[3]}",live=True,preprocess=True),