ali commited on
Commit
6aecff3
·
1 Parent(s): 03a2f57

app.py update

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,6 +1,9 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
 
 
 
4
  # Load your model (update the path to your saved model)
5
  model = pipeline("text-classification", model="./mbti_model_2")
6
 
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ import os
5
+ os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python"
6
+
7
  # Load your model (update the path to your saved model)
8
  model = pipeline("text-classification", model="./mbti_model_2")
9