fixes

andreiionutdamian · Mar 20, 2024 · f2695ed · f2695ed
1 parent f718fed
commit f2695ed
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/app/Dockerfile_monitor b/app/Dockerfile_monitor
@@ -2,7 +2,7 @@ FROM aidamian/base_fastapi:latest
 WORKDIR /test_app
 
 RUN pip3 install torch --index-url https://download.pytorch.org/whl/cpu
-RUN pip install --no-cache transformers scipy psycopg2-binary redis
+RUN pip install --no-cache transformers scipy psycopg2-binary redis pillow
 
 # basic stuff
 ADD src/mixins ./mixins/

diff --git a/app/src/monitor/monitor_app.py b/app/src/monitor/monitor_app.py
@@ -63,8 +63,8 @@ def _update_cache(self, model_type, model_name : str, model):
     result = self.redis_sethash("models", model_type, model_name)
     if result:
       self.P(f"Cache update: {model_type} - {model_name}")
-      labels = model.config.id2label
-      self.P(f"Model labels: {labels}")
+      info = model.config
+      self.P(f"Model config: {info}")
     return result
 
   def set_model(self, model_type: str, model_name: str):