Bugifx for input type

app.py

@@ -51,14 +51,15 @@ def infer(prompt, steps, scale, seed, editing_prompt_1 = None, reverse_editing_d
         del edit_threshold[index]
     editing_prompt_embeddings = None
     
-    
-    if edit_style is not None:
+    out_label = 'SEGA'
+    if edit_style is not None and isinstance(edit_style, str) and edit_style in style_embeddings.keys():
         editing_prompt = None
         reverse_editing_direction = reverse_editing_direction_style
         edit_warmup_steps = edit_warmup_steps_style
         edit_guidance_scale = edit_guidance_scale_style
         edit_threshold = edit_threshold_style
         editing_prompt_embeddings = style_embeddings[edit_style]
+        out_label = edit_style
     
     gen.manual_seed(seed)
     images.extend(pipe(prompt, guidance_scale=scale, num_inference_steps=steps, generator=gen,
@@ -67,7 +68,7 @@ def infer(prompt, steps, scale, seed, editing_prompt_1 = None, reverse_editing_d
                         edit_momentum_scale=edit_momentum_scale, edit_mom_beta=edit_mom_beta
                       ).images)
 
-    return zip(images, ['Original', edit_style if edit_style is not None else 'SEGA'])
+    return zip(images, ['Original', out_label])
 
 def reset_style():
     radio = gr.Radio(label='Style', choices=['Concept Art', 'Animation', 'Character Design', 'Portrait Photo', 'Architecture'])