Update main.py

main.py

@@ -16,7 +16,7 @@ app.add_middleware(  # add the middleware
 def hello():
     return "Hello, I'm Artist"
 
-@app.post("/generate")
+@app.post("/generate/{message}")
 def generate_completion(message: str):
-    response = Linlada._create_completion(model='gpt-4', messages=[{"role": message.role, "content": message.content}], stream=False)
+    response = Linlada._create_completion(model='gpt-4', messages=[{"role": 'user', "content": message}], stream=False)
     return response