Update app.py

app.py

@@ -27,7 +27,7 @@ def find_similarity(base64_image, text_input):
     text_tokens = clip.tokenize([text_input]).to(device)
 
     # Encode image and text features
-    with torch.no_grad():
+    with torch no grad():
         image_features = model.encode_image(image)
         text_features = model.encode_text(text_tokens)
 
@@ -37,7 +37,10 @@ def find_similarity(base64_image, text_input):
     # Convert each element in the similarity array to Decimal
     similarity_decimal = [Decimal(float(score)) for score in similarity]
 
-    return similarity_decimal
+    # Format Decimal values as floats with specific precision (e.g., 4 decimal places)
+    formatted_similarity = [f'{float(score):.5f}' for score in similarity_decimal]
+
+    return formatted_similarity
 
 # Create a Gradio interface
 iface = gr.Interface(