Update app.py

app.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI, Request
 from llama_cpp import Llama
 from huggingface_hub import hf_hub_download
 import os
+import requests
 os.system("ulimit -l unlimited")
 
 app = FastAPI()
@@ -16,17 +17,22 @@ async def index():
 
 @app.post("/api")
 async def completion(request: Request):
-    data = await request.json()
-    prompt = data["prompt"]
-    mode = data['mode']
-    
-    res = model_l(
-        prompt,
-        temperature=0.6,
-        echo=False,
-        max_tokens=41,
-    )
-    return {"responses": res["choices"]}
+    try:
+        data = await request.json()
+        prompt = data["prompt"]
+        mode = data['mode']
+        if mode == 'f':
+            return requests.post("https://mrali-codexpert-computing.hf.space/api", json={"prompt": prompt}).json()
+        else:
+            res = model_l(
+                prompt,
+                temperature=0.6,
+                echo=False,
+                max_tokens=41,
+            )
+        return {"responses": res["choices"]}
+    except:
+        return {"responses": "Error!"}
 
 if __name__ == "__main__":
     import uvicorn