Spaces:

mlkorra
/

OGBV-Bert

Runtime error

mlkorra commited on Sep 8, 2021

Commit

e11f528

•

1 Parent(s): 0677d87

update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -61,7 +61,7 @@ def app():
     # pd.set_option('max_colwidth',30)
     if st.button("Classify"):
         with st.spinner("Classifying the sentence..."):
-            pred,model,tokenizer = load_model(masked_text)
             st.write(pred)
             if st.button('Visualize attributions'):

     # pd.set_option('max_colwidth',30)
     if st.button("Classify"):
         with st.spinner("Classifying the sentence..."):
+            pred = load_model(masked_text)
             st.write(pred)
             if st.button('Visualize attributions'):