Update app.py

app.py

@@ -49,7 +49,7 @@ TEMPLATE = (
     #
     "<|rating:general|>"
     "{aspect_ratio}"
-    "<|length:long|>"
+    "<|length:medium|>"
     #
     "<copyright></copyright>"
     #
@@ -57,8 +57,8 @@ TEMPLATE = (
     #
     "<general>{subject}"
 )
-QUALITY_TAGS = "original style"
-NEGATIVE_PROMPT = ",(bad quality, worst quality:1.1), lowres, bad anatomy, sketch, jpeg artifacts, ugly, poorly drawn, signature, watermark, bad anatomy, bad hands, bad feet, retro, old, 2000s, 2010s, 2011s, 2012s, 2013s, multiple views, 3d, screencap"
+QUALITY_TAGS = ""
+NEGATIVE_PROMPT = "bad quality, worst quality, lowres, bad anatomy, sketch, jpeg artifacts, ugly, poorly drawn, signature, watermark, bad anatomy, bad hands, bad feet, retro, old, 2000s, 2010s, 2011s, 2012s, 2013s, multiple views, 3d, screencap"
 BAN_TAGS = [
     "photoshop (medium)",
     "clip studio paint (medium)",
@@ -126,18 +126,22 @@ else:
 
 
 def get_aspect_ratio(width: int, height: int) -> str:
-    ar = width / height
+    ar = math.log2(width / height)
 
-    if ar <= 1 / math.sqrt(3):
-        return "<|aspect_ratio:ultra_tall|>"
-    elif ar <= 8 / 9:
+    if ar <= -1.25:
+        return "<|aspect_ratio:too_tall|>"
+    elif ar <= -0.75:
+        return "<|aspect_ratio:tall_wallpaper|>"
+    elif ar <= -0.25:
         return "<|aspect_ratio:tall|>"
-    elif ar < 9 / 8:
+    elif ar < 0.25:
         return "<|aspect_ratio:square|>"
-    elif ar < math.sqrt(3):
+    elif ar < 0.75:
         return "<|aspect_ratio:wide|>"
+    elif ar < 1.25:
+        return "<|aspect_ratio:wide_wallpaper|>"
     else:
-        return "<|aspect_ratio:ultra_wide|>"
+        return "<|aspect_ratio:too_wide|>"
 
 
 @torch.inference_mode