Update app.py

app.py

@@ -77,7 +77,7 @@ def health():
     
 # Chat Completion API
 @app.post("/chat/")
-async def complete(gen:GenModel):
+async def chat(gen:GenModel):
     try:
         messages=[
                 {"role": "system", "content": gen.system},
@@ -114,18 +114,16 @@ async def complete(gen:GenModel):
 
 # Chat Completion API
 @app.get("/generate")
-async def complete(
-    question: str,
-    system: str = "You are an AI assistant.",
-    temperature: float = 0.7,
-    seed: int = 42,
-) -> dict:
+async def generate(gen:GenModel):
+    gen.system: str = "You are an AI assistant."
+    gen.temperature: float = 0.5
+    gen.seed: int = 42
     try:
         st = time()
         output = llm_generate.create_chat_completion(
             messages=[
-                {"role": "system", "content": system},
-                {"role": "user", "content": question},
+                {"role": "system", "content": gen.system},
+                {"role": "user", "content": gen.question},
             ],
             temperature=temperature,
             seed=seed,