wilwork commited on
Commit
fe46f6f
·
verified ·
1 Parent(s): f4bfa5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -9
app.py CHANGED
@@ -12,9 +12,7 @@ def process_input(input_data, input_type):
12
  if input_type == "Text":
13
  return model.encode_text([input_data]) if input_data.strip() else None
14
  elif input_type == "Image":
15
- if isinstance(input_data, str): # If it's a file path
16
- image = Image.open(input_data).convert("RGB")
17
- elif isinstance(input_data, np.ndarray): # If it's a NumPy array (Gradio default)
18
  image = Image.fromarray(input_data)
19
  else:
20
  return None # Invalid input type
@@ -44,13 +42,13 @@ def compute_similarity(input1, input2, input1_type, input2_type):
44
  similarity_score = (embedding1 @ embedding2.T).item()
45
  return f"Similarity Score: {similarity_score:.4f}"
46
 
47
- # Function to toggle input fields dynamically
48
  def update_visibility(input1_type, input2_type):
49
  return (
50
- gr.update(visible=(input1_type == "Text")),
51
- gr.update(visible=(input1_type == "Image")),
52
- gr.update(visible=(input2_type == "Text")),
53
- gr.update(visible=(input2_type == "Image"))
54
  )
55
 
56
  # Gradio UI
@@ -71,7 +69,7 @@ with gr.Blocks() as demo:
71
 
72
  output = gr.Textbox(label="Similarity Score / Error", interactive=False)
73
 
74
- # Toggle visibility of inputs dynamically
75
  input1_type.change(update_visibility, inputs=[input1_type, input2_type],
76
  outputs=[input1_text, input1_image, input2_text, input2_image])
77
  input2_type.change(update_visibility, inputs=[input1_type, input2_type],
 
12
  if input_type == "Text":
13
  return model.encode_text([input_data]) if input_data.strip() else None
14
  elif input_type == "Image":
15
+ if isinstance(input_data, np.ndarray): # If it's a NumPy array (Gradio default)
 
 
16
  image = Image.fromarray(input_data)
17
  else:
18
  return None # Invalid input type
 
42
  similarity_score = (embedding1 @ embedding2.T).item()
43
  return f"Similarity Score: {similarity_score:.4f}"
44
 
45
+ # Function to toggle input fields dynamically and clear previous values
46
  def update_visibility(input1_type, input2_type):
47
  return (
48
+ gr.update(visible=(input1_type == "Text"), value="" if input1_type == "Image" else None),
49
+ gr.update(visible=(input1_type == "Image"), value=None),
50
+ gr.update(visible=(input2_type == "Text"), value="" if input2_type == "Image" else None),
51
+ gr.update(visible=(input2_type == "Image"), value=None)
52
  )
53
 
54
  # Gradio UI
 
69
 
70
  output = gr.Textbox(label="Similarity Score / Error", interactive=False)
71
 
72
+ # Toggle visibility of inputs dynamically and clear values
73
  input1_type.change(update_visibility, inputs=[input1_type, input2_type],
74
  outputs=[input1_text, input1_image, input2_text, input2_image])
75
  input2_type.change(update_visibility, inputs=[input1_type, input2_type],