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 |
)
|