app.py

app.py

@@ -12,6 +12,12 @@ import gradio as gr
 #model, preprocess = clip.load("ViT-B/32", jit=False)
 #display output photo
 
+# check if CUDA available 
+device = "cuda" if torch.cuda.is_available() else "cpu"
+  
+# Load the openAI's CLIP model
+model, preprocess = clip.load("ViT-B/32", device=device, jit=False)
+
 # taking photo IDs
 photo_ids = pd.read_csv("./photo_ids.csv")
 photo_ids = list(photo_ids['photo_id'])
@@ -56,11 +62,6 @@ def find_matches(text_features, photo_features, photo_ids, results_count=4):
   
 def image_search(search_text, search_image, option):
   
-  # check if CUDA available 
-  device = "cuda" if torch.cuda.is_available() else "cpu"
-  
-  # Load the openAI's CLIP model
-  model, preprocess = clip.load("ViT-B/32", device=device, jit=False)
   #model = model.to(device)
   
   # Input Text Query