Update main.py
Browse files
main.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
import os
|
| 2 |
import torch
|
| 3 |
import torchaudio
|
| 4 |
-
from fastapi import FastAPI, UploadFile, File, HTTPException
|
| 5 |
from fastapi.responses import JSONResponse
|
| 6 |
from fastapi.middleware.cors import CORSMiddleware
|
| 7 |
from transformers import (
|
|
@@ -15,6 +15,7 @@ from torch.quantization import quantize_dynamic
|
|
| 15 |
import logging
|
| 16 |
import ffmpeg
|
| 17 |
import tempfile
|
|
|
|
| 18 |
|
| 19 |
# ========== Force HF cache to /tmp ==========
|
| 20 |
os.environ["HF_HOME"] = "/tmp/huggingface"
|
|
|
|
| 1 |
import os
|
| 2 |
import torch
|
| 3 |
import torchaudio
|
| 4 |
+
from fastapi import FastAPI, UploadFile, File, HTTPException, Request
|
| 5 |
from fastapi.responses import JSONResponse
|
| 6 |
from fastapi.middleware.cors import CORSMiddleware
|
| 7 |
from transformers import (
|
|
|
|
| 15 |
import logging
|
| 16 |
import ffmpeg
|
| 17 |
import tempfile
|
| 18 |
+
import datetime
|
| 19 |
|
| 20 |
# ========== Force HF cache to /tmp ==========
|
| 21 |
os.environ["HF_HOME"] = "/tmp/huggingface"
|