Merge pull request #17 from atomashevic/dev

Freeze python package versions to prevent dependency/compatibility issues

R/setup_gpu_modules.R

@@ -15,9 +15,11 @@ setup_gpu_modules <- function()
 
   # Set necessary modules
   modules <- c(
-    "autoawq", "auto-gptq", "optimum"
+    "autoawq==0.2.5", "auto-gptq==0.7.1", "optimum==1.19.1"
   )
 
+# TODO freeze versions of modules to their current versions
+
   # Check for Linux
   if(system.check()$OS == "linux"){
     modules <- c(modules, "llama-cpp-python")

R/setup_modules.R

@@ -15,11 +15,11 @@ setup_modules <- function()
 
   # Set necessary modules
   modules <- c(
-    "accelerate", "llama-index", "nltk",
-    "opencv-python", "pandas", "pypdf",
-    "pytube", "pytz", "qdrant-client",
-    "sentencepiece", "tensorflow", "torch",
-    "torchaudio", "torchvision", "transformers"
+    "accelerate==0.29.3", "llama-index==0.10.30", "nltk==3.8.1",
+    "opencv-python", "pandas==2.1.3", "pypdf==4.0.1",
+    "pytube==15.0.0", "pytz==2024.1", "qdrant-client==1.8.2",
+    "sentencepiece==0.2.0", "tensorflow==2.14.1", "torch==2.1.1",
+    "transformers==4.35.0"
   )
 
   # Determine whether any modules need to be installed

README.md

@@ -123,7 +123,7 @@ Path can be either local or an URL. Here's an example of using a URL of Mona Lis
 ```R
 
 # Image URL or local filepath
-image <- "https://upload.wikimedia.org/wikipedia/commons/thumb/7/76/Leonardo_da_Vinci_-_Mona_Lisa.jpg/401px-Leonardo_da_Vinci_-_Mona_Lisa.jpg"
+image <- 'https://cdn.mos.cms.futurecdn.net/xRqbwS4odpkSQscn3jHECh-650-80.jpg'
 
 # Array of emotion labels
 emotions <- c("excitement", "happiness", "pride", "anger", "fear", "sadness", "neutral")

inst/python/image.py

@@ -84,7 +84,7 @@ def get_text_embeds(labels):
     processor_openai
     model_openai
   except NameError:
-    print("Loading OpenAI CLIP model ... \n This may take a minute.")
+    print("Loading OpenAI CLIP model ...")
     processor_openai = AutoProcessor.from_pretrained("openai/clip-vit-base-patch32")
     model_openai = AutoModel.from_pretrained("openai/clip-vit-base-patch32")
   text_inputs_openai = processor_openai(text=labels, return_tensors='pt', padding=True)

inst/python/video.py

@@ -6,12 +6,12 @@
 import pandas as pd
 from pytube import YouTube
 from transformers import AutoProcessor, AutoModel
-import torch
-from torch import nn
-import torchvision.models as models
-from torchvision import transforms
+# import torch
+# from torch import nn
+# import torchvision.models as models
+# from torchvision import transforms
 import torch.nn.functional as F
-from PIL import Image
+# from PIL import Image
 import time