Update app.py

app.py

@@ -8,6 +8,7 @@ import open_clip
 from PIL import Image
 import requests
 import torch
+import warnings
 
 # import torch.nn.functional as F
 import numpy as np
@@ -1138,6 +1139,10 @@ def prepare(raw_idx, text_embeddings, class_order):
 
     choice_values = list(zip(choice_names, choice_values))
 
+    warnings.warn(choice_values)
+
+    warnings.warn(correct_choice)
+
     next_radio = gr.Radio(
         choices=choice_values,
         interactive=True,
@@ -1188,9 +1193,9 @@ def reroll(raw_idx, text_embeddings, class_order):
 
     choice_values = list(zip(choice_names, choice_values))
 
-    gr.Warning(choice_values)
+    warnings.warn(choice_values)
 
-    gr.Warning(correct_choice)
+    warnings.warn(correct_choice)
 
     next_radio = gr.Radio(
         choices=choice_values,