Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ st.set_page_config(
|
|
| 31 |
|
| 32 |
save_memory = False
|
| 33 |
|
| 34 |
-
@st.
|
| 35 |
def load_model():
|
| 36 |
model_path = hf_hub_download('lllyasviel/ControlNet', 'models/control_sd15_scribble.pth')
|
| 37 |
model = create_model('./models/cldm_v15.yaml').cpu()
|
|
@@ -39,7 +39,7 @@ def load_model():
|
|
| 39 |
model = model.cuda()
|
| 40 |
return model
|
| 41 |
|
| 42 |
-
|
| 43 |
def process(input_image, prompt, a_prompt, n_prompt, num_samples, image_resolution, detect_resolution, ddim_steps, guess_mode, strength, scale, seed, eta):
|
| 44 |
with torch.no_grad():
|
| 45 |
|
|
@@ -89,7 +89,7 @@ def process(input_image, prompt, a_prompt, n_prompt, num_samples, image_resoluti
|
|
| 89 |
# return [255 - detected_map] + results
|
| 90 |
return results
|
| 91 |
|
| 92 |
-
@st.
|
| 93 |
def load_lottieurl(url: str):
|
| 94 |
r = requests.get(url)
|
| 95 |
if r.status_code != 200:
|
|
|
|
| 31 |
|
| 32 |
save_memory = False
|
| 33 |
|
| 34 |
+
@st.experimental_singleton
|
| 35 |
def load_model():
|
| 36 |
model_path = hf_hub_download('lllyasviel/ControlNet', 'models/control_sd15_scribble.pth')
|
| 37 |
model = create_model('./models/cldm_v15.yaml').cpu()
|
|
|
|
| 39 |
model = model.cuda()
|
| 40 |
return model
|
| 41 |
|
| 42 |
+
@st.experimental_singleton
|
| 43 |
def process(input_image, prompt, a_prompt, n_prompt, num_samples, image_resolution, detect_resolution, ddim_steps, guess_mode, strength, scale, seed, eta):
|
| 44 |
with torch.no_grad():
|
| 45 |
|
|
|
|
| 89 |
# return [255 - detected_map] + results
|
| 90 |
return results
|
| 91 |
|
| 92 |
+
@st.experimental_memo
|
| 93 |
def load_lottieurl(url: str):
|
| 94 |
r = requests.get(url)
|
| 95 |
if r.status_code != 200:
|