Update app.py

app.py

@@ -41,7 +41,7 @@ def predict(outputs):
             top_cats.append(cat)
             top_probs.append(prob)
             
-    chart_data = pd.DataFrame(top_probs, columns=top_cats)
+    chart_data = pd.DataFrame([top_probs], columns=top_cats)
     st.bar_chart(chart_data)
 
 tokenizer = RobertaTokenizer.from_pretrained("roberta-large-mnli")
@@ -49,14 +49,14 @@ model = init_model()
         
 st.markdown("### Title")
 
-title = st.text_area("* Enter title (required)", height=20)
+title = st.text_area("*Enter title (required)", height=20)
 
 st.markdown("### Abstract")
 
 abstract = st.text_area("Enter abstract", height=200)
 
 if not title:
-    st.warning("Please fill out so required fields")
+    st.warning("Please fill in required fields")
 else:    
     st.markdown("### Result")
     encoded_input = tokenizer(title + '. ' + abstract, return_tensors='pt', padding=True,