Spaces:
Sleeping
Sleeping
Commit
·
219b2cb
1
Parent(s):
def9607
workers 1
Browse files- Dockerfile +1 -1
- 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"
|
|
|
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 = "
|
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 |
)
|