Update main.py

main.py

@@ -19,7 +19,7 @@ app = FastAPI()
 
 #Zephyr completion
 @app.post("/")
-async def stream(item: str):
+async def stream(item: validation):
     system_prompt = 'Below is an instruction that describes a task. Write a response that appropriately completes the request.'
     prompt =  f"<s>[INST]<<SYS>>\n + {system_prompt} + <</SYS>>\n{item.prompt.strip()}[/INST]"
     return llm(prompt)