Merge branch 'main' of https://huggingface.co/spaces/gronkomatic/openclip-embed

app.py

@@ -2,11 +2,13 @@ from typing import Union
 import gradio as gr
 from numpy import empty
 import open_clip
+# import spaces
 import torch
 import PIL.Image as Image
 
 # Set device to GPU if available
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+print(f"PyTorch Device {device}")
 
 # Load the OpenCLIP model and the necessary preprocessors
 # openclip_model = 'laion/CLIP-ViT-B-32-laion2B-s34B-b79K'
@@ -19,6 +21,7 @@ model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms(
 
 
 # Define function to generate text embeddings
+# @spaces.GPU
 def generate_text_embedding(text_data: Union[str, tuple[str]]) -> list[str]:
     """
     Generate embeddings for text data using the OpenCLIP model.