Update main.py

main.py

@@ -1,33 +1,4 @@
-import os
-from dotenv import load_dotenv
-from transformers import TFBertForSequenceClassification, BertTokenizerFast
-import tensorflow as tf
+from app import app
 
-# Load environment variables from .env file
-load_dotenv()
-
-def load_model(model_name):
-    # Load the TensorFlow model from Hugging Face Hub
-    model = TFBertForSequenceClassification.from_pretrained(model_name, use_auth_token=os.getenv('API_KEY'))
-    return model
-
-def load_tokenizer(model_name):
-    tokenizer = BertTokenizerFast.from_pretrained(model_name, use_auth_token=os.getenv('API_KEY'))
-    return tokenizer
-
-def predict(text, model, tokenizer):
-    inputs = tokenizer(text, return_tensors="tf")
-    outputs = model(**inputs)
-    return outputs
-
-def main():
-    model_name = os.getenv('MODEL_PATH')
-    model = load_model(model_name)
-    tokenizer = load_tokenizer(model_name)
-    # Example usage
-    text = "Sample input text"
-    result = predict(text, model, tokenizer)
-    print(result)
-
-if __name__ == "__main__":
-    main()
+if __name__ == '__main__':
+    app.run(debug=True)