Update main.py

main.py

@@ -7,9 +7,9 @@ from transformers import AutoTokenizer, AutoModel
 import torch
 from torch import Tensor
 import torch.nn.functional as F
-from Allam_Backend_HF import (
-    allam_llm
-    ) 
+# from Allam_Backend_HF import (
+#     allam_llm
+#     ) 
 import os
 os.environ['HF_HOME'] = '/'
 
@@ -59,12 +59,12 @@ def e5_embeddings(query: str = Query(...)):
         raise HTTPException(status_code=500)
 
 
-@app.get("/allam_response")
-def allam_response(query: str = Query(...)):
+# @app.get("/allam_response")
+# def allam_response(query: str = Query(...)):
 
-    result = allam_llm(query)
+#     result = allam_llm(query)
 
-    if result is not None:
-        return result
-    else:
-        raise HTTPException(status_code=500)
+#     if result is not None:
+#         return result
+#     else:
+#         raise HTTPException(status_code=500)