Update app.py
Browse files
app.py
CHANGED
|
@@ -17,7 +17,6 @@ if DEVICE == "auto":
|
|
| 17 |
print(f"[SYSTEM] | Using {DEVICE} type compute device.")
|
| 18 |
|
| 19 |
# Variables
|
| 20 |
-
|
| 21 |
MAX_SEED = 9007199254740991
|
| 22 |
DEFAULT_INPUT = ""
|
| 23 |
DEFAULT_NEGATIVE_INPUT = ""
|
|
@@ -35,6 +34,7 @@ model.load_lora_weights(REPO_WEIGHT, weight_name=WEIGHT, adapter_name=ADAPTER)
|
|
| 35 |
model.set_adapters(ADAPTER, adapter_weights=[0.7])
|
| 36 |
model.to(DEVICE)
|
| 37 |
|
|
|
|
| 38 |
def get_seed(seed):
|
| 39 |
seed = seed.strip()
|
| 40 |
if seed.isdigit():
|
|
@@ -66,6 +66,7 @@ def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, height=
|
|
| 66 |
image_paths = [save_image(img) for img in images]
|
| 67 |
return image_paths
|
| 68 |
|
|
|
|
| 69 |
with gr.Blocks() as main:
|
| 70 |
with gr.Column():
|
| 71 |
input = gr.Textbox(lines=1, value=DEFAULT_INPUT, label="Input")
|
|
|
|
| 17 |
print(f"[SYSTEM] | Using {DEVICE} type compute device.")
|
| 18 |
|
| 19 |
# Variables
|
|
|
|
| 20 |
MAX_SEED = 9007199254740991
|
| 21 |
DEFAULT_INPUT = ""
|
| 22 |
DEFAULT_NEGATIVE_INPUT = ""
|
|
|
|
| 34 |
model.set_adapters(ADAPTER, adapter_weights=[0.7])
|
| 35 |
model.to(DEVICE)
|
| 36 |
|
| 37 |
+
# Functions
|
| 38 |
def get_seed(seed):
|
| 39 |
seed = seed.strip()
|
| 40 |
if seed.isdigit():
|
|
|
|
| 66 |
image_paths = [save_image(img) for img in images]
|
| 67 |
return image_paths
|
| 68 |
|
| 69 |
+
# Initialize
|
| 70 |
with gr.Blocks() as main:
|
| 71 |
with gr.Column():
|
| 72 |
input = gr.Textbox(lines=1, value=DEFAULT_INPUT, label="Input")
|