Updated requirements.txt

apps/inference.py

@@ -11,13 +11,13 @@ def load_model(masked_text,model_name):
 
     model = AutoModelForMaskedLM.from_pretrained(model_name, from_flax=True)
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    tokenizer.save_pretrained('exported_pytorch_model')
-    model.save_pretrained('exported_pytorch_model')
+    # tokenizer.save_pretrained('exported_pytorch_model')
+    # model.save_pretrained('exported_pytorch_model')
     nlp = pipeline('fill-mask', model="exported_pytorch_model")
 
     result_sentence = nlp(masked_text)
 
-    return result_sentence
+    return result_sentence[0]['sequence']
 
 def app():
     st.markdown("<h1 style='text-align: center; color: green;'>RoBERTa Hindi</h1>", unsafe_allow_html=True)

requirements.txt

@@ -1,4 +1,6 @@
 streamlit
 torch
 transformers
-jax
+jax
+jaxlib
+flax