Update app.py

app.py

@@ -25,6 +25,9 @@ st.write("Expand your query by leveraging various models")
 default_value = "بيعت الأسلحة في السوق"
 # sent is the variable holding the user's input
 sent = st.text_area("Input", default_value, height=30)
+
+st.checkbox('Use Graph', value=False)
+
 text_st = sent+ ' <mask>'
 
 pipe = pipeline("fill-mask", tokenizer=tokenizer, model=model)