Update app.py

app.py

@@ -3,18 +3,14 @@ import clip
 import torch
 import gradio as gr
 from PIL import Image
-import os
 import base64
 from io import BytesIO
 
 # Load the CLIP model
 model, preprocess = clip.load("ViT-B/32")
-device = "cuda" if torch.cuda.is_available() else "cpu"
+device = "cuda" if torch.cuda.is available() else "cpu"
 model.to(device).eval()
 
-# Define the Business Listing variable
-Business_Listing = "Air Guide"
-
 def find_similarity(image_base64, text_input):
     # Decode the base64 image string to bytes
     image_bytes = base64.b64decode(image_base64)
@@ -38,10 +34,12 @@ def find_similarity(image_base64, text_input):
 
     return similarity[0, 0]
 
-# Define a Gradio interface
 iface = gr.Interface(
     fn=find_similarity,
-    inputs=["text", gr.inputs.Textbox(lines=3, label="Enter Base64 Image"), "text"],
+    inputs=[
+        gr.inputs.Textbox(lines=3, label="Enter Base64 Image"),
+        gr.inputs.Textbox(lines=3, label="Enter Text"),
+    ],
     outputs="number",
     live=True,
     interpretation="default",