omar87 commited on
Commit
0986dc7
·
1 Parent(s): b12bd2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -7,8 +7,8 @@ import io
7
 
8
  # Function to calculate SSIM between two images
9
  def calculate_ssim(img1, img2):
10
- img1_gray = Image.open(io.BytesIO(img1)).convert("L")
11
- img2_gray = Image.open(io.BytesIO(img2)).convert("L")
12
  return ssim(np.array(img1_gray), np.array(img2_gray))
13
 
14
  # Function to compare two trademarks
@@ -25,8 +25,8 @@ def prevent_trademark_conflict(trademark1, trademark2):
25
  iface = gr.Interface(
26
  fn=prevent_trademark_conflict,
27
  inputs=[
28
- gr.inputs.Image(type="file", label="Trademark Image 1"),
29
- gr.inputs.Image(type="file", label="Trademark Image 2")
30
  ],
31
  outputs="text",
32
  title="Trademark Similarity",
 
7
 
8
  # Function to calculate SSIM between two images
9
  def calculate_ssim(img1, img2):
10
+ img1_gray = img1.convert("L")
11
+ img2_gray = img2.convert("L")
12
  return ssim(np.array(img1_gray), np.array(img2_gray))
13
 
14
  # Function to compare two trademarks
 
25
  iface = gr.Interface(
26
  fn=prevent_trademark_conflict,
27
  inputs=[
28
+ gr.inputs.Image(type="pil", label="Trademark Image 1"),
29
+ gr.inputs.Image(type="pil", label="Trademark Image 2")
30
  ],
31
  outputs="text",
32
  title="Trademark Similarity",