Update main.py

main.py

@@ -2,7 +2,7 @@ from flask import Flask, request, jsonify
 from langchain_community.llms import LlamaCpp
 from sentence_transformers import SentenceTransformer
 from transformers import AutoTokenizer, AutoModel
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, HfHubError
 
 # cosine_similarity
 import torch
@@ -15,16 +15,22 @@ n_gpu_layers = 0
 n_batch = 1024
 
 
-model_path = hf_hub_download(repo_id="repo_name", filename="model_file_name", force_download=True)
-
-llm = LlamaCpp(
-    model_path="Phi-3-mini-4k-instruct-q4.gguf",  # path to GGUF file
-    temperature=0.1,
-    n_gpu_layers=n_gpu_layers,
-    n_batch=n_batch,
-    verbose=True,
-    n_ctx=4096
-)
+try:
+    model_path = hf_hub_download(repo_id="repo_name", filename="model_file_name", force_download=True)
+except HfHubError as e:
+    print(f"Error downloading the model: {e}")
+    model_path = None
+
+# تأكد من أن النموذج تم تنزيله بنجاح
+if model_path:
+    llm = LlamaCpp(
+        model_path=model_path,  # path to GGUF file
+        temperature=0.1,
+        n_gpu_layers=n_gpu_layers,
+        n_batch=n_batch,
+        verbose=True,
+        n_ctx=4096
+    )
 
 model0 = AutoModel.from_pretrained('sentence-transformers/paraphrase-TinyBERT-L6-v2')
 model = SentenceTransformer('sentence-transformers/paraphrase-TinyBERT-L6-v2')