Update app.py

app.py

@@ -1,10 +1,12 @@
 import tensorflow as tf
+from transformers import TFBertForSequenceClassification
 from flask import Flask, request, jsonify
 
 app = Flask(__name__)
 
 # Load the model
-model = tf.keras.models.load_model('Erfan11/Neuracraft')
+model_name = "Erfan11/Neuracraft"
+model = TFBertForSequenceClassification.from_pretrained(model_name, use_auth_token="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")
 
 @app.route('/predict', methods=['POST'])
 def predict():
@@ -26,6 +28,4 @@ def postprocess_predictions(predictions):
     pass
 
 if __name__ == '__main__':
-    app.run(debug=True)
-    ## API Endpoints
-- **POST /predict**: Receives JSON data, returns model predictions.
+    app.run(debug=True)