Spaces:
Runtime error
Runtime error
chironjit45
commited on
Commit
·
cc7efce
1
Parent(s):
dff81e5
modified dockerfile
Browse files- Dockerfile +1 -1
- app.py +1 -1
Dockerfile
CHANGED
|
@@ -5,7 +5,7 @@ USER user
|
|
| 5 |
ENV PATH="/home/user/.local/bin:$PATH"
|
| 6 |
|
| 7 |
WORKDIR /app
|
| 8 |
-
COPY --chown=user ./requirements.txt
|
| 9 |
RUN pip install --no-cache-dir --upgrade -r requirements.txt
|
| 10 |
|
| 11 |
copy --chown=user . /app
|
|
|
|
| 5 |
ENV PATH="/home/user/.local/bin:$PATH"
|
| 6 |
|
| 7 |
WORKDIR /app
|
| 8 |
+
COPY --chown=user ./requirements.txt requirements.txt
|
| 9 |
RUN pip install --no-cache-dir --upgrade -r requirements.txt
|
| 10 |
|
| 11 |
copy --chown=user . /app
|
app.py
CHANGED
|
@@ -36,7 +36,7 @@ def translate_text(input_text):
|
|
| 36 |
model = MTModel(mt_pretrained_model = "shhossain/opus-mt-en-to-bn")
|
| 37 |
#Load saved weights
|
| 38 |
model.model.load_state_dict(torch.load(
|
| 39 |
-
r'
|
| 40 |
))
|
| 41 |
model.eval()
|
| 42 |
|
|
|
|
| 36 |
model = MTModel(mt_pretrained_model = "shhossain/opus-mt-en-to-bn")
|
| 37 |
#Load saved weights
|
| 38 |
model.model.load_state_dict(torch.load(
|
| 39 |
+
r'mt_model.pthC:\Users\chiro\PycharmProjects\PythonProject\text\Translation_EN_To_Ban\mt_model.pth'
|
| 40 |
))
|
| 41 |
model.eval()
|
| 42 |
|