Update app.py

app.py

@@ -44,11 +44,11 @@ def generate(query: Query):
     # Decode & extract assistant response
     output_string = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
     response = output_string.split("assistant")[-1].strip()
-    return response
+    return {"response": response}
 
 @app.get("/")
 def root():
-    return {"debai API is running!"}
+    return {"debai": "API is running!"}
 
 if __name__ == "__main__":
     import uvicorn