Update main.py

main.py

@@ -2,7 +2,7 @@ from fastapi import FastAPI, Request
 from pydantic import BaseModel
 from llama_cpp import Llama
 
-model_path = "meta-llama-3-8b.Q4_K_M.gguf"
+model_path = "meta-llama-3-8b-instruct.Q4_K_M.gguf"
 
 llm = Llama(
     model_path = model_path,
@@ -23,13 +23,13 @@ async def solve_dsa_problem(item: DSAPrompt):
                     If the input is not a DSA problem, politely refuse their request and reinsist to provide a DSA problem.
                     '''
     
-    # res = llm.create_chat_completion(
-    #     messages=[
-    #         {"role": "system", "content": system_prompt},
-    #         {"role": "user", "content": item.prompt}
-    #     ],
-    #     temperature = 0.7,
-    #     max_tokens = 2048,
-    # )
+    res = llm.create_chat_completion(
+        messages=[
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": item.prompt}
+        ],
+        temperature = 0.7,
+        max_tokens = 2048,
+    )
 
-    return llm(item.prompt, temperature = 0.9, max_tokens = 2048)
+    return res