wd-tagger

Runtime error

SmilingWolf commited on Dec 13, 2022

Commit

104e60a

•

1 Parent(s): f6dbb10

Bugfix: set default model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -184,7 +184,7 @@ def main():
         fn=func,
         inputs=[
             gr.Image(type="pil", label="Input"),
-            gr.Radio(["ViT", "ConvNext"], label="Model"),
             gr.Slider(
                 0,
                 1,

         fn=func,
         inputs=[
             gr.Image(type="pil", label="Input"),
+            gr.Radio(["ViT", "ConvNext"], value="ViT", label="Model"),
             gr.Slider(
                 0,
                 1,