Spaces:

oracat
/

PaperClassifier

Sleeping

oracat commited on Apr 16, 2023

Commit

f87eabb

1 Parent(s): 8397f59

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ def process(text):
     """
     pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
     result = pipe(text)[0]
-    return result'label']
 tokenizer, model = prepare_model()

     """
     pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
     result = pipe(text)[0]
+    return result['label']
 tokenizer, model = prepare_model()