Update app.py

app.py

@@ -90,15 +90,15 @@ def get_style_embeddings(style_file):
 import torch
 
 def sharpness_loss(image):
-    threshold = torch.tensor(0.5) 
+    threshold = torch.tensor(0.9) 
     std_dev = torch.std(image, dim=[2, 3])
 
     # Calculate the mean standard deviation across the batch
     mean_std_dev = torch.mean(std_dev, dim=0)
 
     # Check if the mean standard deviation is below the threshold
-    loss = torch.mean(mean_std_dev - threshold)*50
-    return -loss
+    loss = torch.abs(torch.mean(mean_std_dev - threshold))*50
+    return loss
 
 
 from torchvision.transforms import ToTensor
@@ -278,7 +278,7 @@ def image_generator(prompt = "dog", loss_function=None):
 
   return display_images_in_rows(generated_sd_images, titles)
 
-description = "Generate an image with a prompt (takes time)"
+description = "Generate an image with a prompt"
 
 demo = gr.Interface(image_generator,
                     inputs=[gr.Textbox(label="prompt", type="text", value="cat fight"),