Update main.py

main.py

@@ -17,6 +17,6 @@ def hello():
     return "Hello, I'm Artist"
 
 @app.post("/generate")
-def generate_completion(message: Message):
+def generate_completion(message: str):
     response = Linlada._create_completion(model='gpt-4', messages=[{"role": message.role, "content": message.content}], stream=False)
     return response