Update app.py

app.py

@@ -41,7 +41,9 @@ def run_model():
 
     model_image_to_text = torch.argmax(embeddings_matrix, dim=1)
 
-    return text_list[model_image_to_text[0]]
+    label_product = text_list[model_image_to_text[0]]
+
+    return label_product, database[label_product] 
 
 def predict(image):
     # Save the image to the desired file path