WajeehAzeemX commited on
Commit
219b2cb
·
1 Parent(s): def9607
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. app.py +1 -1
Dockerfile CHANGED
@@ -10,4 +10,4 @@ COPY --chown=user ./requirements.txt requirements.txt
10
  RUN pip install --no-cache-dir --upgrade -r requirements.txt
11
 
12
  COPY --chown=user . /app
13
- CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "5"]
 
10
  RUN pip install --no-cache-dir --upgrade -r requirements.txt
11
 
12
  COPY --chown=user . /app
13
+ CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
app.py CHANGED
@@ -15,7 +15,7 @@ print(device)
15
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
16
 
17
  # Load the model and processor
18
- model_id = "WajeehAzeemX/whisper-small-ar2_onnx"
19
  model = ORTModelForSpeechSeq2Seq.from_pretrained(
20
  model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
21
  )
 
15
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
16
 
17
  # Load the model and processor
18
+ model_id = "./whisper-small-ar2_onnx"
19
  model = ORTModelForSpeechSeq2Seq.from_pretrained(
20
  model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
21
  )