Update multiple attempts of downloading

app.py

@@ -24,14 +24,24 @@ except AttributeError:
 repo_id = "TheBloke/openbuddy-mistral-7B-v13-GGUF"
 filename = "openbuddy-mistral-7b-v13.Q4_K_S.gguf"  # 4.17G
 
+model_ready = True
 logger.debug("Start dl")
-try:
-    model_path = hf_hub_download(repo_id=repo_id, filename=filename, revision="main")
-except Exception as exc:
-    logger.error(f"failed to download {filename}: {exc}")
-    raise gr.Error("hf acting up, can't donwload the model, exiting")
-    # raise SystemExit("hf acting up, can't donwload the model, exiting")
-logger.debug("Done dl")
+# try to download 5 times:
+for idx in range(5):
+    logger.debug(f"attempt {idx + 1}")
+    try:
+        model_path = hf_hub_download(repo_id=repo_id, filename=filename, revision="main")
+        break
+    except Exception as exc:
+        logger.error(f"failed to download {filename}: {exc}")
+        # raise SystemExit("hf acting up, can't donwload the model {filename=}, exiting")
+else:
+    logger.warning("Tried 5 times to no vain")
+    raise gr.Error(f"hf acting up, can't donwload the model {filename=}, exiting")
+    # raise SystemExit("hf acting up, can't donwload the model {filename=}, exiting")
+    model_ready = False
+
+logger.debug(f"Done dl, {model_ready=}")
 
 # Set gpu_layers to the number of layers to offload to GPU. Set to 0 if no GPU acceleration is available on your system.
 # model = AutoModelForCausalLM.from_pretrained("TheBloke/openbuddy-mistral-7B-v13-GGUF", model_file="openbuddy-mistral-7b-v13.Q4_K_S.gguf", model_type="mistral", gpu_layers=0)
@@ -51,13 +61,19 @@ else:
     gpu_layers = 0
 
 logger.debug("Start loading the model")
-model = AutoModelForCausalLM.from_pretrained(
-    model_path, model_type="mistral", gpu_layers=gpu_layers
-)
-logger.debug("Done loading the model")
+try:
+    model = AutoModelForCausalLM.from_pretrained(
+        model_path, model_type="mistral", gpu_layers=gpu_layers
+    )
+except Exception as exc:
+    logger.error(exc)
+    model_ready = False
+    model = None
+
+logger.debug(f"Done loading the model, {model_ready=}")
 
 ins = """[INST] <<FRIDAY>>
-Remember that your English name is "Shi-Ci" and your name in Chinese is "兮辞". You are a helpful, respectful and honest assistant. Always answer as helpfully as possible, while being safe.  Your answers should not include any harmful, unethical, racist, sexist, toxic, dangerous, or illegal content. Please ensure that your responses are socially unbiased and positive in nature.
+Remember that your English name is "openbuddy" and your name in Chinese is "开友". You are a helpful, respectful and honest assistant. Always answer as helpfully as possible, while being safe.  Your answers should not include any harmful, unethical, racist, sexist, toxic, dangerous, or illegal content. Please ensure that your responses are socially unbiased and positive in nature.
 If a question does not make any sense, or is not factually coherent, explain why instead of answering something not correct. If you don't know the answer to a question, please don't share false information.
 <</FRIDAY>>
 {} [/INST]
@@ -78,7 +94,10 @@ theme = gr.themes.Monochrome(
 
 
 def response(question):
-    res = model(ins.format(question))
+    if model is None:
+        res = "model not ready (got a problem with downloading the file {filename=} from hf.co)"
+    else:
+        res = model(ins.format(question))
     yield res