From 5015db6edbcd0e93b3b444de291d1c63051cfa06 Mon Sep 17 00:00:00 2001 From: Kshitij Date: Sat, 22 Feb 2025 23:09:33 +0530 Subject: [PATCH] Added dependencies in requirements.txt file. --- models/app.py | 41 +++++++++++++++++++++++++++++++++++++++++ models/requirements.txt | 40 ++++++++++++++++++++++++++++++++++++++++ 2 files changed, 81 insertions(+) create mode 100644 models/app.py create mode 100644 models/requirements.txt diff --git a/models/app.py b/models/app.py new file mode 100644 index 0000000..cfacb75 --- /dev/null +++ b/models/app.py @@ -0,0 +1,41 @@ +# Import libraries +from PIL import Image, UnidentifiedImageError +from transformers import ViTImageProcessor, ViTForImageClassification + +# Specify the local directory where the model files are stored +local_model_path = '/home/overnion/Status200/models/pretrained' + +# Load the image processor and model from the local directory +image_processor = ViTImageProcessor.from_pretrained(local_model_path) +model = ViTForImageClassification.from_pretrained( + local_model_path, + ignore_mismatched_sizes=True +) + +# Load image +try: + image = Image.open('/home/overnion/Status200/models/samples/apple.png') + # Convert the image to RGB if it's not already + if (image.mode != 'RGB'): + image = image.convert('RGB') +except FileNotFoundError: + print("Error: Image file not found.") + exit() +except UnidentifiedImageError: + print("Error: Unable to open image. Check the file type.") + exit() +except Exception as e: + print(f"An error occurred: {e}") + exit() + +# Preparing the image for the model +inputs = image_processor(images=image, return_tensors="pt") + +# Make the prediction +outputs = model(**inputs) +logits = outputs.logits +predicted_class_idx = logits.argmax(-1).item() + +# Print the predicted class +print("Predicted class:", model.config.id2label[predicted_class_idx]) + diff --git a/models/requirements.txt b/models/requirements.txt new file mode 100644 index 0000000..5dc595a --- /dev/null +++ b/models/requirements.txt @@ -0,0 +1,40 @@ +certifi==2025.1.31 +charset-normalizer==3.4.1 +filelock==3.17.0 +fsspec==2025.2.0 +huggingface-hub==0.29.1 +idna==3.10 +Jinja2==3.1.5 +MarkupSafe==3.0.2 +mpmath==1.3.0 +networkx==3.4.2 +numpy==2.2.3 +nvidia-cublas-cu12==12.4.5.8 +nvidia-cuda-cupti-cu12==12.4.127 +nvidia-cuda-nvrtc-cu12==12.4.127 +nvidia-cuda-runtime-cu12==12.4.127 +nvidia-cudnn-cu12==9.1.0.70 +nvidia-cufft-cu12==11.2.1.3 +nvidia-curand-cu12==10.3.5.147 +nvidia-cusolver-cu12==11.6.1.9 +nvidia-cusparse-cu12==12.3.1.170 +nvidia-cusparselt-cu12==0.6.2 +nvidia-nccl-cu12==2.21.5 +nvidia-nvjitlink-cu12==12.4.127 +nvidia-nvtx-cu12==12.4.127 +packaging==24.2 +pillow==11.1.0 +PyYAML==6.0.2 +regex==2024.11.6 +requests==2.32.3 +safetensors==0.5.2 +setuptools==75.8.0 +sympy==1.13.1 +tokenizers==0.21.0 +torch==2.6.0 +torchvision==0.21.0 +tqdm==4.67.1 +transformers==4.49.0 +triton==3.2.0 +typing_extensions==4.12.2 +urllib3==2.3.0