Hugging Face's logo

Spaces:
ktllc
/
Clip-Model-InputBase64-Duplicate_Testing
Sleeping

App Files Community
ktllc commited on
Commit
cdab2a8
·
1 Parent(s): c652364

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from io import BytesIO
8
 
9
  # Load the CLIP model
10
  model, preprocess = clip.load("ViT-B/32")
11
- device = "cuda" if torch.cuda.is available() else "cpu"
12
  model.to(device).eval()
13
 
14
  def find_similarity(image_base64, text_input):
 
8
 
9
  # Load the CLIP model
10
  model, preprocess = clip.load("ViT-B/32")
11
+ device = "cuda" if torch.cuda.is_available() else "cpu"
12
  model.to(device).eval()
13
 
14
  def find_similarity(image_base64, text_input):