Update app.py

app.py

@@ -18,8 +18,6 @@ def load_model():
 @st.cache(suppress_st_warning=True, hash_funcs={tokenizers.Tokenizer: lambda _: None})  
 def predict(title, summary, tokenizer, model):
     text = title + "\n" + summary
-    if len(text) < 20:
-        return 'error'
     tokens = tokenizer.encode(text)
     with torch.no_grad():
         logits = model(torch.as_tensor([tokens]))[0]
@@ -60,7 +58,7 @@ button = st.button('Run')
 if button:
     prediction, prediction_probs = predict(title, summary, tokenizer, model)
     ans = get_results(prediction, prediction_probs)
-    if ans == 'error':
+    if len(title + "\n" + summary) < 20:
         st.error("Your input is too short. It is probably not a real article, please try again.")
     else:
         st.subheader('Results:')