Random genre selection if none has been selected. (#2)

- Random genre selection if none has been selected. (5140c811ae4e6bc3166cd62f75d61e8322346975)


Co-authored-by: Carlos Marí Noguera <[email protected]>

app.py

@@ -44,6 +44,10 @@ def main():
             code = genres_dict[genre]
             cond[code-1] = code
 
+        if torch.any(cond != 0):
+            random_number = torch.randint(0, 13, (1,)).item()
+            cond[random_number] = random_number + 1
+
         def callback(image, progress):
             image = DiffusionImageAPI(None).tensor_to_image(image.squeeze(0))
             img_buffer = io.BytesIO()