supercontrast

supercontrast is a package for easily running machine learning models from a variety of providers in a unified interface. We're adding more tasks and providers all the time, and would love help from the community to add more!

We currently support the following tasks:

• OCR
• Sentiment Analysis
• Transcription
• Translation

From some of the most popular providers:

• AWS
• Azure
• GCP
• OpenAI
• Anthropic
• ...and more!

Installation

pip install supercontrast

NOTE: supercontrast is supported on Python 3.12, it may be unstable on other versions. If you have conda installed, you can create an environment with the required packages using the environment.yml file:

conda env create -f environment.yml

Usage

from supercontrast import (
    Provider,
    SentimentAnalysisRequest,
    SuperContrastClient,
    Task,
)

# Sending a Sentiment Analysis Request to AWS
client = SuperContrastClient(task=Task.SENTIMENT_ANALYSIS, providers=[Provider.AWS])
input_text = "I love programming in Python!"
response, metadata = client.request(SentimentAnalysisRequest(text=input_text))

For more examples of how to use supercontrast, refer to examples.py in the examples folder.

Contributing

We welcome contributions to the project! To contribute, please follow these steps:

1. Clone the repo

git clone https://github.com/supercontrast/supercontrast.git

2. Install package

pip install -e .[dev]

3. Run linting

black .
isort .

4. Run tests

pytest -k <test_name>

5. Submit a PR

Submit a PR to the main branch! We will review and merge your PR.