wilwork commited on
Commit
08f9b31
·
verified ·
1 Parent(s): 2241cd5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -21,15 +21,19 @@ def compute_similarity(input1, input2, type1, type2):
21
  if type1 == "Image":
22
  image1 = Image.open(input1).convert("RGB")
23
  input1_tensor = processor(images=image1, return_tensors="pt")["pixel_values"]
24
- else:
25
  input1_tensor = tokenizer(input1, return_tensors="pt")["input_ids"]
 
 
26
 
27
  # Process input2
28
  if type2 == "Image":
29
  image2 = Image.open(input2).convert("RGB")
30
  input2_tensor = processor(images=image2, return_tensors="pt")["pixel_values"]
31
- else:
32
  input2_tensor = tokenizer(input2, return_tensors="pt")["input_ids"]
 
 
33
 
34
  # Compute embeddings
35
  with torch.no_grad():
 
21
  if type1 == "Image":
22
  image1 = Image.open(input1).convert("RGB")
23
  input1_tensor = processor(images=image1, return_tensors="pt")["pixel_values"]
24
+ elif isinstance(input1, str) and input1.strip():
25
  input1_tensor = tokenizer(input1, return_tensors="pt")["input_ids"]
26
+ else:
27
+ return "Error: Invalid text input for Input 1"
28
 
29
  # Process input2
30
  if type2 == "Image":
31
  image2 = Image.open(input2).convert("RGB")
32
  input2_tensor = processor(images=image2, return_tensors="pt")["pixel_values"]
33
+ elif isinstance(input2, str) and input2.strip():
34
  input2_tensor = tokenizer(input2, return_tensors="pt")["input_ids"]
35
+ else:
36
+ return "Error: Invalid text input for Input 2"
37
 
38
  # Compute embeddings
39
  with torch.no_grad():