AbdoIR commited on
Commit
6be1daa
·
verified ·
1 Parent(s): d82dc46

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
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"