Upload folder using huggingface_hub

image_caption/get_caption.py

@@ -47,6 +47,8 @@ def get_image_caption(img_obj, prompt=None):
         caption = json.loads(response.text)["captions"]
         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.")
     else:
         return "Error in generating the caption!"