Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import transformers
 from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForMaskedLM
 
@@ -29,7 +30,7 @@ default_value = "التاريخ هو تحليل و"
 sent = st.text_area("Text", default_value, height = 60)
 sent += ' <mask>'
 
-pipe = pipeline("fill-mask", tokenizer = tokenizer, model = model, device=0)
+pipe = pipeline("fill-mask", tokenizer = tokenizer, model = model)
 dict_next_words = next_word(sent, pipe)
 
 st.write(dict_next_words)