fix: cant proceed to launch
This commit is contained in:
+1
-1
@@ -19,4 +19,4 @@ ENV HSA_OVERRIDE_GFX_VERSION=10.3.0
|
|||||||
ENV PYTORCH_ROCM_ARCH=gfx1030
|
ENV PYTORCH_ROCM_ARCH=gfx1030
|
||||||
|
|
||||||
# Команда для запуска приложения
|
# Команда для запуска приложения
|
||||||
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "9854", "--log-level", "debug"]
|
CMD ["python3", "app.py"]
|
||||||
|
|||||||
@@ -26,7 +26,7 @@ model = None
|
|||||||
def load_model():
|
def load_model():
|
||||||
global model
|
global model
|
||||||
logger.info("Loading whisper model...")
|
logger.info("Loading whisper model...")
|
||||||
model = whisper.load_model("medium", device="cuda")
|
model = whisper.load_model("medium", device="cuda", in_memory=True)
|
||||||
logger.info("Whisper model loaded.")
|
logger.info("Whisper model loaded.")
|
||||||
|
|
||||||
|
|
||||||
@@ -199,7 +199,7 @@ async def transcribe_audio(
|
|||||||
def main():
|
def main():
|
||||||
import uvicorn
|
import uvicorn
|
||||||
get_keys()
|
get_keys()
|
||||||
|
uvicorn.run(app, host="0.0.0.0", port=9854, log_level="debug")
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
|||||||
Reference in New Issue
Block a user