Update main.py

main.py

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