Spaces:

daspartho
/

anime-or-not

Runtime error

daspartho commited on Sep 6, 2022

Commit

d9f9642

•

1 Parent(s): f655b71

typo fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ model = AutoModelForSequenceClassification.from_pretrained("daspartho/anime-or-n
 label_map = {
     'LABEL_0':"That's not Anime enough!",
-    'LABEL_1':"That's a Anime!",
 }
 pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer)

 label_map = {
     'LABEL_0':"That's not Anime enough!",
+    'LABEL_1':"That's an Anime!",
 }
 pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer)