Update main.py

main.py

@@ -7,13 +7,13 @@ import tensorflow as tf
 MODEL_NAME = "Erfan11/Neuracraft"
 API_KEY = "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"
 
-def load_model(model_name):
+def load_model(Erfan11/Neuracraft):
     # Load the TensorFlow model from Hugging Face Hub
-    model = TFBertForSequenceClassification.from_pretrained(model_name, use_auth_token=API_KEY)
+    model = TFBertForSequenceClassification.from_pretrained(model_name, use_auth_token=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)
     return model
 
-def load_tokenizer(model_name):
-    tokenizer = BertTokenizerFast.from_pretrained(model_name, use_auth_token=API_KEY)
+def load_tokenizer(Erfan11/Neuracraft):
+    tokenizer = BertTokenizerFast.from_pretrained(model_name, use_auth_token=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)
     return tokenizer
 
 def predict(text, model, tokenizer):
@@ -22,8 +22,8 @@ def predict(text, model, tokenizer):
     return outputs
 
 def main():
-    model_name = MODEL_NAME
-    model = load_model(model_name)
+    model_name = Erfan11/Neuracraft
+    model = load_model(Erfan11/Neuracraft)
     tokenizer = load_tokenizer(model_name)
     # Example usage
     text = "Sample input text"