nigeljw commited on
Commit
f93e041
1 Parent(s): 74ccab5

Added accelerate package to requirements

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -0
app.py CHANGED
@@ -71,8 +71,8 @@ def diffuse(prompt, negativePrompt, inputImage, mask, guidanceScale, numInferenc
71
 
72
  prompt = gradio.Textbox(label="Prompt", placeholder="A person in a room", lines=3)
73
  negativePrompt = gradio.Textbox(label="Negative Prompt", placeholder="Text", lines=3)
74
- inputImage = gradio.Image(label="Input Image", type="pil")
75
- #inputImage = gradio.Image(label="Input Feed", source="webcam", shape=[512,512], streaming=True)
76
  mask = gradio.Image(label="Mask", type="pil")
77
  outputImage = gradio.Image(label="Extrapolated Field of View")
78
  guidanceScale = gradio.Slider(label="Guidance Scale", maximum=1, value=0.75)
 
71
 
72
  prompt = gradio.Textbox(label="Prompt", placeholder="A person in a room", lines=3)
73
  negativePrompt = gradio.Textbox(label="Negative Prompt", placeholder="Text", lines=3)
74
+ #inputImage = gradio.Image(label="Input Image", type="pil")
75
+ inputImage = gradio.Image(label="Input Feed", source="webcam", shape=[512,512], streaming=True)
76
  mask = gradio.Image(label="Mask", type="pil")
77
  outputImage = gradio.Image(label="Extrapolated Field of View")
78
  guidanceScale = gradio.Slider(label="Guidance Scale", maximum=1, value=0.75)
requirements.txt CHANGED
@@ -3,5 +3,6 @@ Pillow
3
  torch
4
  torchvision
5
  xformers
 
6
  diffusers
7
  transformers
 
3
  torch
4
  torchvision
5
  xformers
6
+ accelerate
7
  diffusers
8
  transformers