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
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
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
def setUpClass(cls): from app.main import get_pipeline get_pipeline.cache_clear()