AnshuPatel4298 commited on
Commit
974d713
·
1 Parent(s): d81a458

Fix protobuf compatibility issues

Browse files
Files changed (2) hide show
  1. app.py +3 -0
  2. requirements.txt +1 -1
app.py CHANGED
@@ -3,6 +3,9 @@ from transformers import BertTokenizer, TFBertForSequenceClassification
3
  import tensorflow as tf
4
  import numpy as np
5
 
 
 
 
6
  # Paths to your models hosted on Hugging Face
7
  basic_model_url = "https://huggingface.co/anshupatel4298/bert-chatbot-model/resolve/main/basic_chatbot_model.h5"
8
  bert_model_name = "anshupatel4298/bert-chatbot-model/bert_model"
 
3
  import tensorflow as tf
4
  import numpy as np
5
 
6
+ import os
7
+ os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python"
8
+
9
  # Paths to your models hosted on Hugging Face
10
  basic_model_url = "https://huggingface.co/anshupatel4298/bert-chatbot-model/resolve/main/basic_chatbot_model.h5"
11
  bert_model_name = "anshupatel4298/bert-chatbot-model/bert_model"
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
  tensorflow==2.9.0
2
  transformers==4.19.2
3
- protobuf<3.20
4
  streamlit
 
1
  tensorflow==2.9.0
2
  transformers==4.19.2
3
+ protobuf==3.20.1
4
  streamlit