new photos

.gitignore

@@ -1,7 +1,3 @@
-#.model.pth
-
-
-
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]
@@ -28,10 +24,8 @@ wandb/
 
 
 .vscode/
-https://github.com/higumax/sketchKeras-pytorch.git
 
 .startup.sh
-
 startup.sh
 
 

app.py

@@ -82,7 +82,7 @@ with gr.Blocks(theme=theme) as demo:
         with gr.Column():
             seed_slider = gr.Number(
                 label="Random Seed 🎲 (if the image generated is not to your liking, simply use another seed)",
-                value=random.randint(0, 10000),
+                value=5,
             )
 
             upscale_button = gr.Checkbox(

load_model.py

@@ -83,7 +83,7 @@ def sample(sketch, scribbles, sampling_steps, seed_nr, progress):
 
         for t in progress.tqdm(
             noise_scheduler.timesteps,
-            desc="Sampling",
+            desc="Painting 🖌🖌🖌",
         ):
             noise_for_plain = noise_scheduler.scale_model_input(noise_for_plain, t).to(
                 device
@@ -110,4 +110,8 @@ def sample(sketch, scribbles, sampling_steps, seed_nr, progress):
 
     sample = torch.clamp((noise_for_plain / 2) + 0.5, 0, 1)
 
-    return transforms.ToPILImage()(sample[0].cpu())
+    image = transforms.ToPILImage()(sample[0].cpu())
+
+    image.save("results/sample.png")
+
+    return image