Update app.py

app.py

@@ -2,6 +2,19 @@ import torch # the library
 import torch.nn as nn # so we can just write nn.X insted of torch.nn.X
 import gradio as gr
 from PIL import ImageColor
+def parse_color_str(clrstr):
+    if clrstr[0]=='#': return clrstr #looks good already
+    if not clrstr.startswith('rgb'): # neither rgb(r,g,b) nor rgba(r,g,b,a)
+        gr.Warning(f"Unsupported color format: {clrstr}")
+        return "#000000"
+
+    # extract substring inside ( ), then split by comma
+    values=clrstr.split('(')[1].split(')')[0].split(',')
+
+    r = min(max(round(float(values[0].strip())), 0), 255)
+    g = min(max(round(float(values[1].strip())), 0), 255)
+    b = min(max(round(float(values[2].strip())), 0), 255)
+    return (r,g,b)
 class ColorGen(nn.Module):
     def __init__(self,dim,vocab_size):
         super().__init__()
@@ -46,7 +59,7 @@ model.load_state_dict(torch.load("color_gen.pt",map_location=torch.device("cpu")
 
 def get_name(color,temp:float):
     if color is not None:
-        color = ImageColor.getcolor(color,"RGB")
+        color = parse_color_str(color)
     else:
         color = (0,0,0)
     name = "".join([vocab[_] for _ in model.name_color(torch.tensor([color])/255,color_pad_len,temp)]).strip()