From 89364cd049f75fb857e43b7b13b3d2ff72802dae Mon Sep 17 00:00:00 2001
From: Wiktor Walentynowicz <wiktor.walentynowicz@pwr.edu.pl>
Date: Fri, 16 Dec 2022 14:23:12 +0100
Subject: [PATCH] Update predcitions and python version.
---
.gitlab-ci.yml | 2 +-
Dockerfile | 2 +-
Dockerfile.gpu | 2 +-
requirements-gpu.txt | 2 +-
requirements.txt | 1 +
src/winer_worker.py | 2 +-
6 files changed, 6 insertions(+), 5 deletions(-)
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 7eb8988..9dedc36 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -1,4 +1,4 @@
-image: "clarinpl/python:3.8"
+image: "python:3.10"
cache:
paths:
diff --git a/Dockerfile b/Dockerfile
index 446cfc1..4e8a261 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM clarinpl/python:3.8
+FROM python:3.10
COPY requirements.txt requirements.txt
RUN python3 -m pip install -r requirements.txt && rm requirements.txt
diff --git a/Dockerfile.gpu b/Dockerfile.gpu
index 2a53004..ff4cde9 100644
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.7.0-cudnn8-runtime-ubuntu20.04
+FROM nvidia/cuda:11.7.0-cudnn8-runtime-ubuntu22.04
RUN DEBIAN_FRONTEND=noninteractive apt-get update && apt-get install -y gcc python3-dev python3-venv python3-pip
diff --git a/requirements-gpu.txt b/requirements-gpu.txt
index cca044f..3508634 100644
--- a/requirements-gpu.txt
+++ b/requirements-gpu.txt
@@ -1,4 +1,4 @@
--index-url https://pypi.clarin-pl.eu/simple/
nlp_ws
-winer==0.2.0
+winer==0.3.0
awscli==1.22.57
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 02e500a..3508634 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
--index-url https://pypi.clarin-pl.eu/simple/
nlp_ws
+winer==0.3.0
awscli==1.22.57
\ No newline at end of file
diff --git a/src/winer_worker.py b/src/winer_worker.py
index 7dae252..f676237 100644
--- a/src/winer_worker.py
+++ b/src/winer_worker.py
@@ -26,7 +26,7 @@ class WinerWorker:
) -> None:
documents = [create_document_from_clarin_json(read_clarin_json(input_path))]
outputs = self.active_model.predict(
- [document.get_pretokenized_text() for document in documents]
+ [str(document) for document in documents]
)
for idx in range(len(documents)):
--
GitLab