app.py

app.py

@@ -9,7 +9,7 @@ from io import BytesIO
 import requests
 import gradio as gr
 # Load the openAI's CLIP model
-model, preprocess = clip.load("ViT-B/32", jit=False)
+#model, preprocess = clip.load("ViT-B/32", jit=False)
 #display output photo
 def show_output_image(matched_images) :
    image=[]
@@ -48,8 +48,8 @@ def image_search(search_text, search_image, option):
   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)
+  model, preprocess = clip.load("ViT-B/32", device=device, jit=False)
+  #model = model.to(device)
   
   # Input Text Query
   #search_query = "The feeling when your program finally works"