Python cache_clear Exemples

Exemple #1

Fichier : test_api_image_classification.py Projet : muellerzr/huggingface_hub

    def setUp(self):
        self.old_model_id = os.getenv("MODEL_ID")
        self.old_task = os.getenv("TASK")
        os.environ["MODEL_ID"] = self.model_id
        os.environ["TASK"] = "image-classification"
        from app.main import app, get_pipeline

        get_pipeline.cache_clear()

        self.app = app

Exemple #2

    def setUp(self):
        self.old_model_id = os.getenv("MODEL_ID")
        self.old_task = os.getenv("TASK")
        os.environ["MODEL_ID"] = self.model_id
        os.environ["TASK"] = "automatic-speech-recognition"

        from app.main import app, get_pipeline

        get_pipeline.cache_clear()

        self.app = app

Exemple #3

Fichier : test_api_audio_to_audio.py Projet : muellerzr/huggingface_hub

    def setUp(self):
        self.old_model_id = os.getenv("MODEL_ID")
        self.old_task = os.getenv("TASK")
        os.environ["MODEL_ID"] = self.model_id
        os.environ["TASK"] = "audio-to-audio"

        from app.main import app, get_pipeline

        get_pipeline.cache_clear()

        self.app = app

Exemple #4

    def setUpClass(cls):
        from app.main import get_pipeline

        get_pipeline.cache_clear()