Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -79,8 +79,8 @@ def upload_image_to_r2(image, account_id, access_key, secret_key, bucket_name):
|
|
79 |
aws_secret_access_key=secret_key
|
80 |
)
|
81 |
|
82 |
-
current_time = datetime.now().strftime("%Y/%m/%d/%H
|
83 |
-
image_file = f"generated_images/{current_time}
|
84 |
buffer = BytesIO()
|
85 |
image.save(buffer, "PNG")
|
86 |
buffer.seek(0)
|
@@ -108,7 +108,6 @@ def generate_random_4_digit_string():
|
|
108 |
return ''.join(random.choices(string.digits, k=4))
|
109 |
|
110 |
@spaces.GPU(duration=120)
|
111 |
-
|
112 |
def run_lora(prompt, image_url, lora_strings_json, image_strength, cfg_scale, steps, randomize_seed, seed, width, height, upload_to_r2, account_id, access_key, secret_key, bucket, progress=gr.Progress(track_tqdm=True)):
|
113 |
print("run_lora", prompt, lora_strings_json, cfg_scale, steps, width, height)
|
114 |
gr.Info("Starting process")
|
|
|
79 |
aws_secret_access_key=secret_key
|
80 |
)
|
81 |
|
82 |
+
current_time = datetime.now().strftime("%Y/%m/%d/%H/%M/%S")
|
83 |
+
image_file = f"generated_images/{current_time}/{random.randint(0, MAX_SEED)}.png"
|
84 |
buffer = BytesIO()
|
85 |
image.save(buffer, "PNG")
|
86 |
buffer.seek(0)
|
|
|
108 |
return ''.join(random.choices(string.digits, k=4))
|
109 |
|
110 |
@spaces.GPU(duration=120)
|
|
|
111 |
def run_lora(prompt, image_url, lora_strings_json, image_strength, cfg_scale, steps, randomize_seed, seed, width, height, upload_to_r2, account_id, access_key, secret_key, bucket, progress=gr.Progress(track_tqdm=True)):
|
112 |
print("run_lora", prompt, lora_strings_json, cfg_scale, steps, width, height)
|
113 |
gr.Info("Starting process")
|