Upload folder using huggingface_hub

image_caption/get_caption.py

@@ -59,7 +59,9 @@ def get_image_caption(img_obj, prompt=None):
         print(f"Caption = {caption}")
         return caption
     elif response.status_code == 502:
-        print("Model is getting loaded, please wait for 3-4 minutes and then try again.")
+        print("Status 502")
+        return "Model is getting loaded, please wait for 3-4 minutes and then try again."
+
     else:
         return "Error in generating the caption!"