Update app.py

app.py

@@ -125,10 +125,15 @@ def predict(images, input_text, model_name):
     clip_text_time = time.time() - start_time
 
     start_time = time.time()
-    results = client.query(embedding_input=text_embeddings[0].tolist())
+    try:
+        results = client.query(embedding_input=text_embeddings[0].tolist())
+        output = ''
+    except:
+        results = []
+        output = 'The server for image retrieval is not working. Please try again later.'
     retrieval_time = time.time() - start_time
 
-    output = ''
+
 
     for idx, result in enumerate(results):
         image_url = result['url']