Spaces:
Paused
Paused
Added default prompts and fixed inference pause control
Browse files
app.py
CHANGED
@@ -65,7 +65,7 @@ def Slerp(start, end, alpha):
|
|
65 |
second = torch.sin(alpha*omega)/sinOmega
|
66 |
return first.unsqueeze(1)*start + second.unsqueeze(1)*end
|
67 |
|
68 |
-
def Diffuse(latentWalk, generatorSeed, inputImage, mask, prompt, negativePrompt, guidanceScale, numInferenceSteps):
|
69 |
global lastImage, lastSeed, generator, oldLatentWalk, activeLatents
|
70 |
|
71 |
if mask is None or pauseInference is True:
|
@@ -130,8 +130,8 @@ with gradio.Blocks(live=True) as ux:
|
|
130 |
#staticLatents = gradio.Checkbox(label="Static Latents", info=staticLatentsDesc, value=True, interactive=True)
|
131 |
inputImage = gradio.Image(label="Input Feed", source="webcam", shape=[512,512], streaming=True)
|
132 |
mask = gradio.Image(label="Mask", type="pil", value=defaultMask)
|
133 |
-
prompt = gradio.Textbox(label="Prompt", info=promptDesc, placeholder=examplePrompt1, lines=3)
|
134 |
-
negativePrompt = gradio.Textbox(label="Negative Prompt", info=negPromptDesc, placeholder="Facial hair", lines=3)
|
135 |
guidanceScale = gradio.Slider(label="Guidance Scale", info="A higher value causes the generation to be more relative to the text prompt conditioning.", maximum=100, minimum=1, value=7.5, step= 0.1)
|
136 |
numInferenceSteps = gradio.Slider(label="Number of Inference Steps", info=numInfStepsDesc, maximum=100, minimum=1, value=20, step=1)
|
137 |
generatorSeed = gradio.Slider(label="Generator Seed", info=generatorSeedDesc, maximum=10000, minimum=1, value=lastSeed, step=1)
|
@@ -145,7 +145,7 @@ with gradio.Blocks(live=True) as ux:
|
|
145 |
outputImage = gradio.Image(label="Extrapolated Field of View")
|
146 |
pauseInference = gradio.Checkbox(label="Pause Inference", value=False)
|
147 |
|
148 |
-
inferenceInputs = [latentWalk, generatorSeed, inputImage, mask, prompt, negativePrompt, guidanceScale, numInferenceSteps]
|
149 |
generateLatents.click(GenerateNewLatentsForInference, outputs=latentWalk)
|
150 |
inputImage.change(fn=Diffuse, inputs=inferenceInputs, outputs=outputImage, show_progress=False)
|
151 |
|
@@ -165,4 +165,4 @@ with gradio.Blocks(live=True) as ux:
|
|
165 |
#ux = gradio.Interface(fn=diffuse, title="View Diffusion", article=article, description=description, inputs=inputs, outputs=outputImage, examples=inputExamples, live=True)
|
166 |
|
167 |
print("Launching Demo")
|
168 |
-
ux.launch()
|
|
|
65 |
second = torch.sin(alpha*omega)/sinOmega
|
66 |
return first.unsqueeze(1)*start + second.unsqueeze(1)*end
|
67 |
|
68 |
+
def Diffuse(latentWalk, generatorSeed, inputImage, mask, prompt, negativePrompt, guidanceScale, numInferenceSteps, pauseInference):
|
69 |
global lastImage, lastSeed, generator, oldLatentWalk, activeLatents
|
70 |
|
71 |
if mask is None or pauseInference is True:
|
|
|
130 |
#staticLatents = gradio.Checkbox(label="Static Latents", info=staticLatentsDesc, value=True, interactive=True)
|
131 |
inputImage = gradio.Image(label="Input Feed", source="webcam", shape=[512,512], streaming=True)
|
132 |
mask = gradio.Image(label="Mask", type="pil", value=defaultMask)
|
133 |
+
prompt = gradio.Textbox(label="Prompt", info=promptDesc, placeholder=examplePrompt1, value="A person", lines=3)
|
134 |
+
negativePrompt = gradio.Textbox(label="Negative Prompt", info=negPromptDesc, placeholder="Facial hair", value="Text, words", lines=3)
|
135 |
guidanceScale = gradio.Slider(label="Guidance Scale", info="A higher value causes the generation to be more relative to the text prompt conditioning.", maximum=100, minimum=1, value=7.5, step= 0.1)
|
136 |
numInferenceSteps = gradio.Slider(label="Number of Inference Steps", info=numInfStepsDesc, maximum=100, minimum=1, value=20, step=1)
|
137 |
generatorSeed = gradio.Slider(label="Generator Seed", info=generatorSeedDesc, maximum=10000, minimum=1, value=lastSeed, step=1)
|
|
|
145 |
outputImage = gradio.Image(label="Extrapolated Field of View")
|
146 |
pauseInference = gradio.Checkbox(label="Pause Inference", value=False)
|
147 |
|
148 |
+
inferenceInputs = [latentWalk, generatorSeed, inputImage, mask, prompt, negativePrompt, guidanceScale, numInferenceSteps, pauseInference]
|
149 |
generateLatents.click(GenerateNewLatentsForInference, outputs=latentWalk)
|
150 |
inputImage.change(fn=Diffuse, inputs=inferenceInputs, outputs=outputImage, show_progress=False)
|
151 |
|
|
|
165 |
#ux = gradio.Interface(fn=diffuse, title="View Diffusion", article=article, description=description, inputs=inputs, outputs=outputImage, examples=inputExamples, live=True)
|
166 |
|
167 |
print("Launching Demo")
|
168 |
+
ux.launch() #debug=True)
|