Update main.py

main.py

@@ -2,26 +2,26 @@ 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, HfHubError
+from huggingface_hub import hf_hub_download, HfApi
 
 # cosine_similarity
 import torch
 from torch.nn.functional import cosine_similarity
-import os 
+import os
 
 app = Flask(__name__)
 
 n_gpu_layers = 0
 n_batch = 1024
 
-
+# تنزيل النموذج باستخدام معالجة الأخطاء
 try:
     model_path = hf_hub_download(repo_id="repo_name", filename="model_file_name", force_download=True)
-except HfHubError as e:
+except Exception 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
@@ -76,5 +76,12 @@ def compare():
     index = sim.index(max_sim)
     return jsonify({'job': jobs_skills[index]})
 
+@app.route('/models', methods=['GET'])
+def list_models():
+    hf_api = HfApi()
+    models = hf_api.list_models()
+    return jsonify({'models': models})
+
 if __name__ == '__main__':
     app.run()
+