kaitongg's picture
Update README.md
e577b9c verified
metadata
license: mit
datasets:
  - kaitongg/Critique_text-dataset

Best Architect Text Predictor

This is an AutoGluon model trained to classify text data. It was trained on a dataset of haiku and their critiques.

Model

The model is a TabularPredictor trained with AutoGluon, utilizing sentence embeddings from all-MiniLM-L6-v2.

How to use

You can use this model in a Hugging Face Space to input text and get classification predictions.

Files

  • app.py: The Gradio application script to run the Space.
  • requirements.txt: List of required Python packages.
  • AutogluonModels/ag-20250927_223330/: Directory containing the saved AutoGluon predictor and its components.

Inference API

You can use this model with the Hugging Face Inference API.